题目链接:
https://leetcode-cn.com/explore/interview/card/top-interview-quesitons-in-2018/266/heap-stack-queue/1153/
因为栈的先进后出特性,所以我们需要一直将最小值放在栈顶,然而还要维持正常的出进栈,用一个栈完成这样的操作复杂度会在O(n)以上,所以我的思路是用两个栈来实现,一个用来保存正常的操作,一个用来保存当前的最小值。
我们申请两个栈,
- s1用来保存正常的进出栈,需要入栈的时候直接调用push操作,出栈直接调用pop
- s2专门用来保存当前的最小值
-
在s1入栈时判断下面两个条件,如果成立, 那么就入栈,不成立s2就不进行操作,最小值还是s2栈顶的元素
- 1-> s2为空,即当前要入栈的值就是最小值
- 2-> s2的值小于当前栈顶的元素,即保持栈顶是当前所有元素中的最小值
-
在s1出栈时判断,如果当前要出栈的元素和s2栈顶的最小值相等,那么就把这个最小值去掉,变成前一个最小值在栈顶。
-
C++ AC代码:
class MinStack {
public:
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
s1.push(x);
//入栈条件: 1:为空
// 2:元素小于s2的栈顶元素
if(s2.empty()||x<=s2.top())
{
s2.push(x);
}
}
void pop() {
//先让s2出栈,因为s1的栈顶要先拿来对比
if(s2.top()==s1.top())
{
s2.pop();
}
s1.pop();
}
int top() {
return s1.top();
//s1栈顶就是入栈时的顺序
}
int getMin() {
return s2.top();
//s2的栈顶永远保存着当前的最小值
}
private:
stack<int> s1,s2; //申请两个栈
};