首页 > itarticle > map sum

map sum

admin 11月 12, 2020 0

Map Sum

实现一个MapSum类，含有insert和sum 两个方法

python

class MapSum(object):
	def __init__(self):
		self.map={}

	def insert(self,key,val):
		self.map[key]=val

	def sum(self,prefix):
		return sum(val for key,val in self.map.items() if key.startswith(prefix))

java

class MapSum{
	HashMap<String,Integer>map;
	public MapSum(){
		map=new HashMap<String,Integer>();
	}	
	public void insert(String key,int val){
		map.put(key,val);
	}
	public int sum(String prefix){
		int ans=0;
		for(String key:map.keySet()){
			if(key.startsWith(prefix)){
				ans+=map.get(key);
			}
		}
		return ans;
	}
}