美文网首页
Map Sum Pairs

Map Sum Pairs

作者: Frank_Kivi | 来源:发表于2018-06-27 10:42 被阅读0次

https://www.lintcode.com/problem/map-sum-pairs/description

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

class MapSum {
    private Map<String, Integer> map = new HashMap<>();

    /**
     * Initialize your data structure here.
     */
    public MapSum() {

    }

    public void insert(String key, int val) {
        map.put(key, val);
    }

    public int sum(String prefix) {
        int sum = 0;
        Iterator<Map.Entry<String, Integer>> iterator = map.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry<String, Integer> next = iterator.next();
            if (next.getKey().startsWith(prefix)) {
                sum += next.getValue();
            }
        }
        return sum;
    }
}

/**
 * Your MapSum object will be instantiated and called as such:
 * MapSum obj = new MapSum();
 * obj.insert(key,val);
 * int param_2 = obj.sum(prefix);
 */

相关文章

网友评论

      本文标题:Map Sum Pairs

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