-
Notifications
You must be signed in to change notification settings - Fork 0
/
155. 最小栈.cc
69 lines (57 loc) · 1.3 KB
/
155. 最小栈.cc
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
class MinStack {
stack<int> sta_;
multiset<int> uset_;
public:
/** initialize your data structure here. */
MinStack() {}
void push(int x) {
sta_.push(x);
uset_.insert(x);
}
void pop() {
int x = sta_.top();
sta_.pop();
auto iter = uset_.find(x);
uset_.erase(iter);
}
int top() {
return sta_.top();
}
int getMin() {
return *uset_.begin();
}
};
/**
* Your MinStack object will be instantiated and called as such:
* MinStack* obj = new MinStack();
* obj->push(x);
* obj->pop();
* int param_3 = obj->top();
* int param_4 = obj->getMin();
*/
/* 双栈: 36ms */
class MinStack {
stack<int> elemSta;
stack<int> minSta;
public:
/** initialize your data structure here. */
MinStack() {}
void push(int x) {
elemSta.push(x);
if(minSta.empty() || minSta.top() >= x) minSta.push(x);
}
void pop() {
//assert(!elemSta.empty());
int x = elemSta.top();
if(!minSta.empty() && minSta.top() == x) minSta.pop();
elemSta.pop();
}
int top() {
//assert(!elemSta.empty());
return elemSta.top();
}
int getMin() {
//assert(!minSta.empty());
return minSta.top();
}
};