forked from leetcoders/LeetCode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMinStack.h
44 lines (40 loc) · 996 Bytes
/
MinStack.h
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
/*
Author: King, [email protected]
Date: Nov 14, 2014
Problem: Min Stack
Difficulty: Easy
Source: https://oj.leetcode.com/problems/min-stack/
Notes:
Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.
push(x) -- Push element x onto stack.
pop() -- Removes the element on top of the stack.
top() -- Get the top element.
getMin() -- Retrieve the minimum element in the stack.
*/
class MinStack {
public:
void push(int x) {
elements.push(x);
if (minstk.empty() || x <= minstk.top()) {
minstk.push(x);
}
}
void pop() {
if (elements.empty()) {
return;
}
if (elements.top() == minstk.top()) {
minstk.pop();
}
elements.pop();
}
int top() {
return elements.top();
}
int getMin() {
return minstk.top();
}
private:
stack<int> elements;
stack<int> minstk;
};