美文网首页
键值映射

键值映射

作者: xialu | 来源:发表于2021-11-14 23:08 被阅读0次

    来源:力扣(LeetCode)
    链接:https://leetcode-cn.com/problems/map-sum-pairs

    题目描述:

    实现一个 MapSum 类,支持两个方法,insert 和 sum:
    MapSum() 初始化 MapSum 对象
    void insert(String key, int val) 插入 key-val 键值对,字符串表示键 key ,整数表示值 val 。如果键 key 已经存在,那么原来的键值对将被替代成新的键值对。
    int sum(string prefix) 返回所有以该前缀 prefix 开头的键 key 的值的总和。

    示例:

    输入:
    ["MapSum", "insert", "sum", "insert", "sum"]
    [[], ["apple", 3], ["ap"], ["app", 2], ["ap"]]
    输出:
    [null, null, 3, null, 5]

    解释:

    MapSum mapSum = new MapSum();
    mapSum.insert("apple", 3);
    mapSum.sum("ap"); // return 3 (apple = 3)
    mapSum.insert("app", 2);
    mapSum.sum("ap"); // return 5 (apple + app = 3 + 2 = 5)

    代码实现:
    class MapSum {
        int[][] tr = new int[2510][26];
        int[] hash = new int[2510];
        int idx;
        public void insert(String key, int val) {
            int p = 0;
            for (int i = 0; i < key.length(); i++) {
                int u = key.charAt(i) - 'a';
                if (tr[p][u] == 0) tr[p][u] = ++idx;
                p = tr[p][u];
            }
            hash[p] = val;
        }
        public int sum(String prefix) {
            int p = 0;
            for (int i = 0; i < prefix.length(); i++) {
                int u = prefix.charAt(i) - 'a';
                if (tr[p][u] == 0) return 0;
                p = tr[p][u];
            }
            return dfs(p);
        }
        int dfs(int p) {
            int ans = hash[p];
            for (int u = 0; u < 26; u++) {
                if (tr[p][u] != 0) ans += dfs(tr[p][u]);
            }
            return ans;
        }
    }
    
    

    相关文章

      网友评论

          本文标题:键值映射

          本文链接:https://www.haomeiwen.com/subject/qqrftrtx.html