leetcode155最小栈c++实现

题目描述

设计一个支持 push，pop，top 操作，并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。

代码实现

用一个辅助栈的top存储当前最小的值，空间换时间

class MinStack {
public: 
    /** initialize your data structure here. */
    stack<int> a;
    stack<int> b;
    MinStack() {
        
    }
void push(int x) {
    a.push(x);
    if(b.empty() || b.top()>=x)
        b.push(x);
    else b.push(b.top());
}

void pop() {
    a.pop();
    b.pop();
}

int top() {
    return a.top();
}

int getMin() {
    return b.top();
}
};

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack* obj = new MinStack();
 * obj->push(x);
 * obj->pop();
 * int param_3 = obj->top();
 * int param_4 = obj->getMin();
 */