最小栈的实现
思路
- 两个栈,左边栈接受元素,右边栈存最小的元素
- 入栈时,先入左边栈,随后进行比较,左边和右边栈顶元素进行比较,如果新元素小,就把新元素放在右边的栈顶位置,如果新元素大,则还是把右边栈顶元素放在栈顶的位置
- 出栈时,出的是最后一个入栈元素,所以就是左边的栈出栈
- 出最小值时,就是右边的栈顶元素出栈
- 出栈时两个都得出
class MinStack {
public:
MinStack() {
}
void push(int x) {
left.push(x);
int min;
if (!right.empty() && right.top() <= x){
min = right.top();
}
else
{
min = x;
}
right.push(min);
}
void pop() {
left.pop();
right.pop();
}
int top() {
return left.top();
}
int getMin() {
return right.top();
}
stack<int> left;
stack<int> right;
};
优化版本
class MinStack {
public:
MinStack() {
}
void push(int x) {
left.push(x);
if (right.empty() || x <= right.top()){
right.push(x);
}
}
void pop() {
int top = left.top();
left.pop();
if (top == right.top()){
right.pop();
}
}
int top() {
return left.top();
}
int getMin() {
return right.top();
}
stack<int> left;
stack<int> right;
};