题目描述
设计一个支持push,pop,top等操作并且可以在O(1)时间内检索出最小元素的堆栈。
push(x)–将元素x插入栈中
pop()–移除栈顶元素
top()–得到栈顶元素
getMin()–得到栈中最小元素
样例
MinStack minStack = new MinStack();
minStack.push(-1);
minStack.push(3);
minStack.push(-4);
minStack.getMin(); --> Returns -4.
minStack.pop();
minStack.top(); --> Returns 3.
minStack.getMin(); --> Returns -1.
算法
(单调栈)
push()
操作其实无所谓前后,但是pop()操作有,因为存在一个比较top()的操作,如果先pop()了,比较的就不一样了。- get_min和top都要记得return,因为是int型的返回类型。
时间复杂度
$O(1)$
C++ 代码
class MinStack {
public:
stack<int> minStk;
stack<int> stk;
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
// stk.push(x);
if(minStk.empty() || x < minStk.top())
minStk.push(x);
stk.push(x);
}
void pop() {
//stk.pop();
if(stk.top() == minStk.top())
minStk.pop();
stk.pop();
}
int top() {
return stk.top();
}
int getMin() {
return minStk.top();
}
};
/**
* 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();
*/