题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
解题思路:本题考虑到使用两个栈进行一系列操作,使用第一个栈进行push,pop操作,使用第二个栈辅助进行min操作。
具体操作如下(下文用s1表示第一个栈,s2表示第二个栈):
1、push操作:s1.push
2、pop操作:s1.pop
3、min操作:在s1进行push操作时,同时对s2进行push,若本次push进s1的元素值小于s2的栈顶元素,则将该元素同时push进s2。若本次push进s1的元素值大于s2的栈顶元素,将s2的栈顶元素再次push进s2。通俗的说就是构建一个与s1相同高度的栈,但是第n层元素表示从0-n层中最小的元素。进行min操作时则返回s2的栈顶元素。
class MinStack {
public:
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
s1.push(x);
//栈为空或栈顶元素大于插入元素时,将x插入s2中
if(s2.size()==0||x<s2.top())
s2.push(x);