剑指 Offer 30. 包含min函数的栈
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
思路:对于本题而言,重点在于如何维护栈的最小值,即对于一个栈在pop和push后依然能够动态的更新栈内的最小值,这是本题的关键所在,为解决这个问题,应当引入第二个栈,此栈作为最小值栈,即此栈的元素按非严格递增排列(从顶到底)。
①当入栈时,若入栈的元素小于最小值栈的栈顶元素(或最小值栈无元素),则栈1与最小值栈都将此元素入栈。若入栈元素大于最小值栈的栈顶元素,则在栈1入栈,同时将最小值栈的栈顶元素再一次入栈。
②对于pop函数,只需要同时弹出栈1与最小值栈的元素即可,保持同步性。
③对于min函数,在最小值栈取栈顶元素即可。
④top函数,在栈1中取得top函数即可。
代码
class MinStack {
stack<int> stack1;
stack<int> minstack;
public:
// int min_ans = 65535;
/** initialize your data structure here. */
MinStack() {
minstack.push(INT_MAX);
}
void push(int x) {
// stack1.push(x);
// if(x<stack1.top()){
// minstack.push(x);
// }else{
// minstack.push(minstack.top());
// }
if(minstack.empty()||minstack.top()>x){
minstack.push(x);
}else{
minstack.push(minstack.top());
}
stack1.push(x);
}
void pop() {
// stack1.pop();
stack1.pop();
minstack.pop();
}
int top() {
return stack1.top();
}
int min() {
return minstack.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->min();
*/