-
Notifications
You must be signed in to change notification settings - Fork 3
/
432.cpp
89 lines (77 loc) · 2.24 KB
/
432.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
class AllOne {
private:
struct Node {
int value;
unordered_set<string> keys;
};
list<Node> l;
unordered_map<string, list<Node>::iterator> mp;
public:
AllOne() {
}
void inc(string key) {
auto it = mp.find(key);
// not exist
if (it == mp.end()) {
// original list does not have "1"
if (l.empty() || l.front().value != 1) {
l.push_front({1, {key}});
}
// original list has "1"
else {
l.front().keys.insert(key);
}
// mp
mp[key] = l.begin();
return;
}
// exist
auto nodeIt = it->second;
auto nextNodeIt = next(nodeIt);
// original list does not have next
if (nextNodeIt == l.end() || nextNodeIt->value != nodeIt->value + 1) {
nextNodeIt = l.insert(nextNodeIt, {nodeIt->value + 1, {}});
}
nextNodeIt->keys.insert(key);
// mp
mp[key] = nextNodeIt;
// clear list
nodeIt->keys.erase(key);
if (nodeIt->keys.empty()) l.erase(nodeIt);
}
void dec(string key) {
auto it = mp.find(key);
if (it == mp.end()) return;
auto nodeIt = it->second;
if (nodeIt->value > 1) {
auto prevNodeIt = prev(nodeIt);
if (nodeIt == l.begin() || prevNodeIt->value != nodeIt->value - 1) {
prevNodeIt = l.insert(nodeIt, {nodeIt->value - 1, {}});
}
prevNodeIt->keys.insert(key);
mp[key] = prevNodeIt;
}
else {
mp.erase(key);
}
// clear list
nodeIt->keys.erase(key);
if (nodeIt->keys.empty()) l.erase(nodeIt);
}
string getMaxKey() {
if (l.empty()) return "";
return *(l.back().keys.begin());
}
string getMinKey() {
if (l.empty()) return "";
return *(l.front().keys.begin());
}
};
/**
* Your AllOne object will be instantiated and called as such:
* AllOne* obj = new AllOne();
* obj->inc(key);
* obj->dec(key);
* string param_3 = obj->getMaxKey();
* string param_4 = obj->getMinKey();
*/