题目描述
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
示例:
输入:
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]
输出:
[null,null,null,null,-3,null,0,-2]
解释:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
提示:
pop、top 和 getMin 操作总是在 非空栈 上调用。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-stack
解题思路
用两个栈来实现
第一个栈stk,用于正常的对栈的操作
第二个栈minStack,用于保证此栈的栈顶元素一直是最小元素
MinStack函数中:初始化minStack最小栈
push函数中:
stk栈正常放入
minStack栈要与当前栈顶元素(即当前最小元素)做对比,取更小的一个放入
pop函数:正常出栈。
top函数:正常返回栈顶元素
getMin函数:返回minStack中的栈顶元素,即最小元素
代码
class MinStack {
public:
stack<int> stk; //正常栈
stack<int> minStack; //栈顶元素为最小元素的栈
/** initialize your data structure here. */
MinStack() {
minStack.push(INT_MAX); //初始化最小栈
}
void push(int val) {
stk.push(val); //stk正常存入
minStack.push(min(val,minStack.top())); //minStack需要保证栈顶元素始终最小,所以需要对比后再存入
}
void pop() {
stk.pop(); //stk正常弹出
minStack.pop(); //minStack需要同步弹出
}
int top() {
return stk.top();//正常取栈顶元素
}
int getMin() {
return minStack.top();//此处为最小元素
}
};
/**
* Your MinStack object will be instantiated and called as such:
* MinStack* obj = new MinStack();
* obj->push(val);
* obj->pop();
* int param_3 = obj->top();
* int param_4 = obj->getMin();
*/