设计一个 map ,满足以下几点:
字符串表示键,整数表示值
返回具有前缀等于给定字符串的键的值的总和
实现一个 MapSum 类:
MapSum() 初始化 MapSum 对象
void insert(String key, int val) 插入 key-val 键值对,字符串表示键 key ,整数表示值 val 。如果键 key 已经存在,那么原来的键值对 key-value 将被替代成新的键值对。
int sum(string prefix) 返回所有以该前缀 prefix 开头的键 key 的值的总和。
示例 1:
输入:
[“MapSum”, “insert”, “sum”, “insert”, “sum”]
[[], [“apple”, 3], [“ap”], [“app”, 2], [“ap”]]
输出:
[null, null, 3, null, 5]
解释:
MapSum mapSum = new MapSum();
mapSum.insert(“apple”, 3);
mapSum.sum(“ap”); // 返回 3 (apple = 3)
mapSum.insert(“app”, 2);
mapSum.sum(“ap”); // 返回 5 (apple + app = 3 + 2 = 5)
这一题感觉博主写的非常的顺哈,一遍就过了,其实是第二次接触这个算法,只能说这个算法真的很棒,在解决相关问题上,感兴趣的可以学习一下,解题代码如下:
typedef struct {struct MapSum *alphtree[26];bool exitence;int val;} MapSum;MapSum* mapSumCreate() {MapSum *m=(MapSum*)malloc(sizeof(MapSum));for(int i=0;i<26;i++){m->alphtree[i]=NULL;}m->exitence=false;return m;}void mapSumInsert(MapSum* obj, char * key, int val) {for(int i=0;key[i]!='\0';i++){if(obj->alphtree[key[i]-'a']==NULL){obj->alphtree[key[i]-'a']=mapSumCreate();}obj=obj->alphtree[key[i]-'a'];}obj->val=val;obj->exitence=true;}
int sum;
void dfs(MapSum* obj){if(obj->exitence==true){sum=sum+obj->val;}for(int i=0;i<26;i++){if(obj->alphtree[i]){dfs(obj->alphtree[i]);}}}int mapSumSum(MapSum* obj, char * prefix) {for(int i=0;prefix[i]!='\0';i++){if(obj->alphtree[prefix[i]-'a']==NULL){return 0;}obj=obj->alphtree[prefix[i]-'a'];}sum=0;dfs(obj);return sum;}void mapSumFree(MapSum* obj) {free(obj);}/*** Your MapSum struct will be instantiated and called as such:* MapSum* obj = mapSumCreate();* mapSumInsert(obj, key, val);* int param_2 = mapSumSum(obj, prefix);* mapSumFree(obj);
*/