-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy path146.cpp
43 lines (39 loc) · 1.11 KB
/
146.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
class LRUCache {
private:
unordered_map<int, pair<int, list<int>::iterator>> cacheObj;
list<int> LRUObj;
int capacity_;
public:
LRUCache(int capacity) {
capacity_ = capacity;
}
void useItem(unordered_map<int, pair<int, list<int>::iterator>>::iterator& iter) {
LRUObj.erase(iter->second.second);
LRUObj.push_front(iter->first);
iter->second.second = LRUObj.begin();
}
int get(int key) {
auto iter = cacheObj.find(key);
if (iter == cacheObj.end()) {
return -1;
}
useItem(iter);
return iter->second.first;
}
void put(int key, int value) {
// If the key exists
auto iter = cacheObj.find(key);
if (iter != cacheObj.end()) {
useItem(iter);
cacheObj[key] = {value, LRUObj.begin()};
return;
}
if (capacity_ == cacheObj.size()) {
cacheObj.erase(LRUObj.back());
LRUObj.pop_back();
}
LRUObj.push_front(key);
cacheObj.insert({key, {value, LRUObj.begin()}});
return;
}
};