LeetCode677.键值映射（Java实现）

链接：https://leetcode-cn.com/problems/map-sum-pairs/

	class MapSum {
		private Node root;
	    public MapSum() {
	        root=new Node();
	    }
	    
	    public void insert(String word, int val) {
	    	Node cur=root;
			for(int i=0;i<word.length();i++){
				char c=word.charAt(i);
				if(cur.next.get(c)==null){
					cur.next.put(c, new Node());
				}
				cur=cur.next.get(c);
			}
			cur.value=val;
	    }
	    
	    public int sum(String prefix) {
	    	int sum;
	        Node cur=root;
	        for(int i=0;i<prefix.length();i++){
	        	char c=prefix.charAt(i);
	        	if(cur.next.get(c)==null){
	        		return 0;
	        	}
	        	cur=cur.next.get(c);
	        }
	        return sum(cur);
	    }
	    private int sum(Node node){

	    	int res=node.value;
	    	for(char c:node.next.keySet()){
	    		res+=sum(node.next.get(c));
	    	}
	    	return res;
	    }
	    private class Node{
			public TreeMap<Character,Node> next;
			public int value;
			public Node(int value){
				this.value=value;
				next=new TreeMap<Character, Node>();
			}
			public Node(){
				this(0);
			}
		}
	}