677. 鍵值映射

實現(xiàn)一個 MapSum 類，支持兩個方法，insert?和?sum：

MapSum() 初始化 MapSum 對象
void insert(String key, int val) 插入 key-val 鍵值對，字符串表示鍵 key ，整數(shù)表示值 val 。如果鍵 key 已經(jīng)存在，那么原來的鍵值對將被替代成新的鍵值對。
int sum(string prefix) 返回所有以該前綴 prefix 開頭的鍵 key 的值的總和。

示例：輸入： ["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"); // return 3 (apple = 3) mapSum.insert("app", 2); mapSum.sum("ap"); // return 5 (apple + app = 3 + 2 = 5)

提示：

• 1 <= key.length, prefix.length <= 50
• key 和 prefix 僅由小寫英文字母組成
• 1 <= val <= 1000
• 最多調(diào)用 50 次 insert 和 sum

解題思路

使用字典樹維護插入的鍵值對

新建Node類，用來構(gòu)建字典樹，每個節(jié)點擁有26個子節(jié)點，對應(yīng)26個字母，若某個節(jié)點為空，則代表不存在包含該字母的前綴。在每個節(jié)點中記錄，滿足當(dāng)前節(jié)點前綴的總和的大小。

對于前綴 prefix 開頭的鍵 key 的值的總和的處理，每次在字典樹上插入節(jié)點的時候，我們都在插入路徑的每一個節(jié)點上增加val，而替換key值的時候，我們先正常插入key值，當(dāng)發(fā)現(xiàn)key值已經(jīng)被插入過了，就沿著插入路徑給每個節(jié)點減去val。

代碼

class Node { public:Node* child[26] ;int sum,val;bool is_end; };class MapSum { public:Node *root;MapSum() {root=new Node();}void insert(string key, int val) {Node *cur = root;for (int i = 0; i < key.size(); ++i) {if (cur->child[key[i] - 'a'] == nullptr)cur->child[key[i] - 'a'] = new Node();cur = cur->child[key[i] - 'a'];if (cur != nullptr)(cur->sum) += val;}if (cur->is_end) {Node *new_cur = root;for (int i = 0; i < key.size(); ++i) {new_cur = new_cur->child[key[i] - 'a'];if (new_cur != nullptr)(new_cur->sum) -= cur->val;}}cur->is_end=true;cur->val = val;}int sum(string prefix) {Node *cur=root;int sum(0);int i = 0;for (; i < prefix.size()&&cur!= nullptr; ++i) {cur=cur->child[prefix[i]-'a'];if (cur!= nullptr)sum=cur->sum;}return cur!= nullptr?sum:0;} };/*** Your MapSum object will be instantiated and called as such:* MapSum* obj = new MapSum();* obj->insert(key,val);* int param_2 = obj->sum(prefix);*/

總結(jié)

以上是生活随笔為你收集整理的677. 键值映射的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。