因为要O(1)的时间复杂度,所以必须要辅助栈。这里我们用两个栈来实现。s1存放实际入栈元素,s2存放该层及以下最小的元素
push操作是灵魂,s1每次都要push,s2如果一开始为空,也要push。后面如果不为空了,就要进行判断,push进来的值和栈顶的值哪个小,如果小于就push进来。反之再次push进栈顶的元素(方便pop)
pop就是两个栈都进行pop,因为s2每一层存的都是s1对应层及以下最小的元素
top是取实际栈顶,返回s1.top
min返回s2.top
class Solution {
public:
//栈的底层是只能在一端操作的双端队列
//直接用两个stack来实现即可,因为min要o(1)的时复,所以必须要用辅助栈
//pop为什么不会影响取最小值得自己画图看看
void push(int value) {
stack1.push(value);
if(stack2.empty()){
stack2.push(value);
}
else if(stack2.top() >= value) {
stack2.push(value);
}
else{
stack2.push(stack2.top());
}
}
void pop() {
stack1.pop();
stack2.pop();
}
int top() {
return stack1.top();
}
int min() {
return stack2.top();
}
private:
stack<int> stack1, stack2;
};