leetcode 677. 键值映射 python

leetcode 677. 键值映射 python

题目描述：

题解：

1.init函数中初始化一个dict，保存[key,value]键值对。

2.insert函数中，将输入的[key,value]加入或者更新dict。

3.sum函数中初始化一个ssum=0，对dict中每个key进行判断，如果和prefix相符合，则将value加入ssum，返回ssum。

class MapSum(object):

    def __init__(self):
        self.dict = {}


    def insert(self, key, val):
        if key in self.dict.keys():
            self.dict[key] = val
        self.dict[key] = val


    def sum(self, prefix):
        self.ssum = 0
        for eachprefix in self.dict.keys():
            if eachprefix[0:len(prefix)]==prefix:
                self.ssum = self.ssum+self.dict[eachprefix]
        return self.ssum