用 JavaScript 映射求和对
问题
我们需要实现一个 MapSum 类,该类具有 insert 和 sum 方法。对于 insert 方法,我们将得到一对 (字符串、整数)。字符串表示键,整数表示值。如果键已存在,则将原来的键值对覆盖为新的键值对。
对于 sum 方法,我们将得到一个表示前缀的字符串,我们需要返回所有键以该前缀开头的键值对的值得和。
示例
以下是代码 −
class Node { constructor(val) { this.num = 0 this.val = val this.children = {} } } class MapSum { constructor(){ this.root = new Node(''); } } MapSum.prototype.insert = function (key, val) { let node = this.root for (const char of key) { if (!node.children[char]) { node.children[char] = new Node(char) } node = node.children[char] } node.num = val } MapSum.prototype.sum = function (prefix) { let sum = 0 let node = this.root for (const char of prefix) { if (!node.children[char]) { return 0 } node = node.children[char] } const helper = (node) => { sum += node.num const { children } = node Object.keys(children).forEach((key) => { helper(children[key]) }) } helper(node) return sum } const m = new MapSum(); console.log(m.insert('apple', 3)); console.log(m.sum('ap'));
输出
undefined 3
广告