JAVA程序设计：键值映射（LeetCode：677）

实现一个 MapSum 类里的两个方法，insert 和 sum。

对于方法 insert，你将得到一对（字符串，整数）的键值对。字符串表示键，整数表示值。如果键已经存在，那么原来的键值对将被替代成新的键值对。

对于方法 sum，你将得到一个表示前缀的字符串，你需要返回所有以该前缀开头的键的值的总和。

示例 1:

输入: insert("apple", 3), 输出: Null
输入: sum("ap"), 输出: 3
输入: insert("app", 2), 输出: Null
输入: sum("ap"), 输出: 5

思路：采用前缀树（字典树），唯一的坑点是若插入相同的键的话需要处理下val，具体操作可以看我代码！

class MapSum {

	class node{
		node[] a;
		int val;
		public node(int val) {
			this.a=new node[26];
			for(int i=0;i<26;i++)
				this.a[i]=null;
			this.val=val;
		}
	}
	
	private node root;
	private Map<String,Integer> map;
	
    /** Initialize your data structure here. */
    public MapSum() {
    	root=new node(0);
    	map=new HashMap<>();
    }
    
    public void insert(String key, int val) {
    	node myRoot=root;
    	if(map.containsKey(key))
    		val=val-map.get(key);
    	map.put(key, val);
        for(int i=0;i<key.length();i++) {
        	int tmp=key.charAt(i)-'a';
        	if(myRoot.a[tmp]==null)
        		myRoot.a[tmp]=new node(val);
        	else
        		myRoot.a[tmp].val+=val;
        	//System.out.println(myRoot.val);
        	myRoot=myRoot.a[tmp];
        }
    }
    
    public int sum(String prefix) {
        node myRoot=root;
        for(int i=0;i<prefix.length() && myRoot!=null;i++) {
        	int tmp=prefix.charAt(i)-'a';
        	myRoot=myRoot.a[tmp];
        }
        return myRoot==null?0:myRoot.val;
    }
}