- 题目描述
设计一个支持 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.
- 打卡
(1) GTA 2021-03-11 打卡
(2) LeetCode 2021-03-11 题155提交记录
(3) 运行结果
- 思路
设计一个类,给这个类两个成员变量:stack s,int minValue。
当进行push操作的,先算最小值(考虑当前s为空的时候,就等于压进来的值,当不空的时候,用压的值和最小值比较即可);然后再对s做push操作。
取top()值,直接用s.top()即可。
getMin(),直接返回成员变量minValue。
pop()操作稍微复杂一点。需要考虑pop出的值是否为最小值,如果不是最小值,则直接s.pop()即可;当pop()的值为最小值的时候,需要借助一个 stack tmp来计算剩下的元素的最小值。具体方法是:先初始化minValue = INT_MAX,然后pop原来的栈的元素。接下来需要把s中的元素都放到tmp中,然后再从tmp中放回s中(放回s后的顺序会和原来一致)。再放回s的过程中,可以类似push操作的时候,再重新算minValue。
- 代码
class MinStack {
public:
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
minValue = s.empty()? x : min(x, minValue);
s.push(x);
}
void pop() {
if(minValue == s.top())
{
minValue = INT_MAX;
s.pop();
stack<int> tmp;
while(not s.empty())
{
tmp.push(s.top());
s.pop();
}
while(not tmp.empty())
{
s.push(tmp.top());
minValue = s.empty()? tmp.top() : min(tmp.top(), minValue);
tmp.pop();
}
}
else
{
s.pop();
}
}
int top() {
return s.top();
}
int getMin() {
return minValue;
}
private:
stack<int> s;
int minValue;
};