leetcode hot100 之 最小栈

题目

设计一个支持 push ，pop ，top *** 作，并能在常数时间内检索到最小元素的栈。

实现 MinStack 类:

MinStack() 初始化堆栈对象。
void push(int val) 将元素val推入堆栈。
void pop() 删除堆栈顶部的元素。
int top() 获取堆栈顶部的元素。
int getMin() 获取堆栈中的最小元素。

输入：
[“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“top”,“getMin”]
[[],[-2],[0],[-3],[],[],[],[]]
输出：
[null,null,null,null,-3,null,0,-2]
解释：
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

原题链接：https://leetcode.cn/problems/min-stack/

思路

维护一个辅助栈，大小和栈保持一样，栈顶是最小值即可。假如新加入一个元素，如果这个元素比栈顶大，则更新栈顶的最小值即可。

• 复杂度分析
  • 时间复杂度 O(1)。所有 *** 作都是 O(1)
  • 空间复杂度 O(n)。存储 n 个元素时，两个栈都是 O(n)

代码

class MinStack {
public:
    MinStack() {
    }
    
    void push(int val) {
        ori_stack.push(val);
        if (min_stack.empty() || val < min_stack.top()) {
            min_stack.push(val);
        }
        else {
            min_stack.push(min_stack.top());
        }
    }
    
    void pop() {
        ori_stack.pop();
        min_stack.pop();
    }
    
    int top() {
        return ori_stack.top();
    }
    
    int getMin() {
        return min_stack.top();
    }
private:
    stack<int> ori_stack;
    stack<int> min_stack;
};

/**
 * 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();
 */