forked from wisdompeak/LeetCode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path1670.Design-Front-Middle-Back-Queue_v2.cpp
96 lines (89 loc) · 2.04 KB
/
1670.Design-Front-Middle-Back-Queue_v2.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
90
91
92
93
94
95
96
class FrontMiddleBackQueue {
list<int> List;
list<int>::iterator mid;
int n = 0;
public:
FrontMiddleBackQueue() {
}
void pushFront(int val)
{
List.push_front(val);
if (n==0)
mid = List.begin();
else if (n%2==1)
mid = prev(mid);
n++;
}
void pushMiddle(int val)
{
if (n==0)
{
List.push_back(val);
mid = List.begin();
}
else if (n%2==0)
{
List.insert(next(mid), val);
mid = next(mid);
}
else
{
List.insert(mid, val);
mid = prev(mid);
}
n++;
}
void pushBack(int val)
{
List.push_back(val);
if (n==0)
mid = List.begin();
else if (n%2==0)
mid = next(mid);
n++;
}
int popFront()
{
if (n==0) return -1;
int ret = List.front();
if (n%2==0)
mid = next(mid);
List.pop_front();
n--;
return ret;
}
int popMiddle()
{
if (n==0) return -1;
int ret = *mid;
list<int>::iterator mid_new;
if (n%2==1)
mid_new = prev(mid);
else
mid_new = next(mid);
List.erase(mid);
n--;
mid = mid_new;
return ret;
}
int popBack()
{
if (n==0) return -1;
int ret = List.back();
if (n%2==1)
mid = prev(mid);
List.pop_back();
n--;
return ret;
}
};
/**
* Your FrontMiddleBackQueue object will be instantiated and called as such:
* FrontMiddleBackQueue* obj = new FrontMiddleBackQueue();
* obj->pushFront(val);
* obj->pushMiddle(val);
* obj->pushBack(val);
* int param_4 = obj->popFront();
* int param_5 = obj->popMiddle();
* int param_6 = obj->popBack();
*/