155. 最小栈

1.题目介绍

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

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

示例 1:
输入:
["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.

2.题解

2.1 辅助栈

注意下在空栈时,往最小栈填充一个无限大,防止空栈时调用getMin();

思路

使用两个栈数据结构来存储,一个用来存储栈数据,一个同步存储栈中最小数据,和栈同步push,pop

代码

class MinStack {
public:
    MinStack() {
        minStk.push(INT_MAX);
    }
    
    void push(int val) {
        stk.push(val);
        minStk.push(min(val, minStk.top()));    
    }
    
    void pop() {
        stk.pop();
        minStk.pop();
    }
    
    int top() {
        return stk.top();
    }
    
    int getMin() {
        return minStk.top();
    }
private:
    stack<int> stk;
    stack<int> minStk;
};

/**
 * 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();
 */
posted @ 2024-01-22 17:43  DawnTraveler  阅读(5)  评论(0编辑  收藏  举报