155. 最小栈(中等)
方法一:辅助栈
思路
- 题目要求在常数时间内获得栈中的最小值,因此不能在的时候再去计算getMin0最小值,最好应该在 push 或者 pop 的时候就已经计算好了当前栈中的最小值。
- 我在 push 和 pop 操作都设置了「辅助栈」。St 用来保存正常输入顺序的数字, minSt 在 栈 temp 的帮助下,始终按照从小到大的顺序(即
minSt.top())
是最小元素)排列。 - 这种方法下,
St
和minSt
的数据始终是一样的。因此并不是一个好的思路,请看方法二。
代码
class MinStack {
public:
stack<int> St, minSt;
MinStack() {
}
void push(int val) {
St.push(val);
stack<int> temp;
if(minSt.empty()){
minSt.push(val);
}
else{
while(!minSt.empty()){
if(minSt.top() >= val){
minSt.push(val);
break;
}
else{
temp.push(minSt.top());
minSt.pop();
}
}
while(!temp.empty()){
minSt.push(temp.top());
temp.pop();
}
}
}
void pop() {
int head = St.top();
St.pop();
stack<int> temp;
while(!minSt.empty()){
if(minSt.top() != head){
temp.push(minSt.top());
minSt.pop();
}
else{
minSt.pop();
break;
}
}
while(!temp.empty()){
minSt.push(temp.top());
temp.pop();
}
}
int top() {
return St.top();
}
int getMin() {
return minSt.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();
*/
方法二:辅助栈优化
思路
-
方法一用到了三个栈,其中
St
和minSt
的数据始终是一样的,当数据栈的数越来越多的时候,我们要在辅助栈minSt
的栈顶放置和 栈St
栈顶一样的元素,这样做有点“浪费”。基于这一点,我们做一些“优化”。
代码
class MinStack {
public:
stack<int> St, minSt;
MinStack() {
minSt.push(INT_MAX);
}
void push(int val) {
St.push(val);
if(val <= minSt.top()){
minSt.push(val);
}else{
minSt.push(minSt.top());
}
}
void pop() {
St.pop();
minSt.pop();
}
int top() {
return St.top();
}
int getMin() {
return minSt.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();
*/