-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathAbstract Classes - Polymorphism
More file actions
111 lines (107 loc) · 2.37 KB
/
Abstract Classes - Polymorphism
File metadata and controls
111 lines (107 loc) · 2.37 KB
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#include <iostream>
#include <vector>
#include <map>
#include <string>
#include <algorithm>
#include <set>
#include <cassert>
using namespace std;
struct Node{
Node* next;
Node* prev;
int value;
int key;
Node(Node* p, Node* n, int k, int val):prev(p),next(n),key(k),value(val){};
Node(int k, int val):prev(NULL),next(NULL),key(k),value(val){};
};
class Cache{
protected:
map<int,Node*> mp; //map the key to the node in the linked list
int cp; //capacity
Node* tail; // double linked list tail pointer
Node* head; // double linked list head pointer
virtual void set(int, int) = 0; //set function
virtual int get(int) = 0; //get function
};
class LRUCache: Cache {
public:
LRUCache(int l) {
cp = l;
tail = NULL;
head = NULL;
mp = {};
}
virtual void set(int k, int v) {
auto it = mp.find(k);
if (it == mp.end()) {
Node *node = new Node(k, v);
if (cp == 0) {
node->next = head;
head->prev = node;
head = node;
Node *tmp = tail;
mp.erase(tmp->key);
tail = tail->prev;
delete tmp;
} else {
cp--;
if (head == NULL) {
head = node;
tail = node;
} else {
node->next = head;
head->prev = node;
head = node;
}
}
mp[k] = node;
} else {
Node *tmp = it->second;
if (tmp != head) {
if (tmp == tail) {
tail = tmp->prev;
tmp->prev->next = NULL;
tmp->prev = NULL;
tmp->next = head;
head->prev = tmp;
head = tmp;
} else {
tmp->prev->next = tmp->next;
tmp->next->prev = tmp->prev;
tmp->prev = NULL;
tmp->next = head;
head->prev = tmp;
head = tmp;
}
}
head->value = v;
}
}
virtual int get(int k) {
auto it = mp.find(k);
if (it != mp.end()) {
return it->second->value;
}
return -1;
}
};
int main() {
int n, capacity,i;
cin >> n >> capacity;
LRUCache l(capacity);
for(i=0;i<n;i++) {
string command;
cin >> command;
if(command == "get") {
int key;
cin >> key;
cout << l.get(key) << endl;
}
else if(command == "set") {
int key, value;
cin >> key >> value;
l.set(key,value);
}
}
return 0;
}