文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

請你實現一個數據結構支持以下操作：

• Inc(key) - 插入一個新的值為 1 的 key。
  或者使一個存在的 key 增加一，保證 key 不為空字符串。
• Dec(key) - 如果這個 key 的值是 1，那么把他從數據結構中移除掉。
  否則使一個存在的 key 值減一。
  如果這個 key 不存在，這個函數不做任何事情。key 保證不為空字符串。
• GetMaxKey() - 返回 key 中值最大的任意一個。
  如果沒有元素存在，返回一個空字符串"" 。
• GetMinKey() - 返回 key 中值最小的任意一個。
  如果沒有元素存在，返回一個空字符串""。

挑戰：
你能夠以 O(1) 的時間復雜度實現所有操作嗎？

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/all-oone-data-structure
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

參考大佬的題解

class node { public:int val;unordered_set<string> s;node(int v){val = v;} }; class AllOne {unordered_map<string, list<node>::iterator> m;list<node> l; public:/** Initialize your data structure here. */AllOne() {}/** Inserts a new key <Key> with value 1. Or increments an existing key by 1. */void inc(string key) {auto it = m.find(key);if(it == m.end()){if(l.empty() || l.front().val > 1)l.push_front(node(1));//沒有val為1的，新建l.begin()->s.insert(key);m[key] = l.begin();}else{auto iter = it->second;//迭代器位置int num = iter->val;//當前數字iter->s.erase(key);//原位置處刪除auto olditer = iter;auto newiter = ++iter;num++;if(newiter != l.end() && newiter->val == num){ //新位置數字剛好是+1以后的newiter->s.insert(key);m[key] = newiter;}else{ //新開辟節點auto temp = l.insert(newiter,node(num));//新節點之前插入temp->s.insert(key);//返回的當前節點存入字符串m[key] = temp;}if(olditer->s.empty())l.erase(olditer);//原來節點為空要刪除}}/** Decrements an existing key by 1. If Key's value is 1, remove it from the data structure. */void dec(string key) {auto it = m.find(key);if(it == m.end()) return;auto iter = it->second;int num = iter->val;iter->s.erase(key);auto olditer = iter;auto newiter = --iter;num--;if(num == 0)m.erase(key);else if(olditer != l.begin() && newiter->val == num){ //前面還有節點，且數字吻合newiter->s.insert(key);m[key] = newiter;}else{ //前面數字不符合，在老節點前插入auto temp = l.insert(olditer,node(num));temp->s.insert(key);m[key] = temp;}if(olditer->s.empty())l.erase(olditer);}/** Returns one of the keys with maximal value. */string getMaxKey() {if(l.empty()) return "";return *(l.back().s.begin());}/** Returns one of the keys with Minimal value. */string getMinKey() {if(l.empty()) return "";return *(l.front().s.begin());} };

100 ms 24.9 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 432. 全 O(1) 的数据结构（设计题）*的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。