155. Min Stack(最小栈)

版权声明:转载请注明出处 https://blog.csdn.net/qq_37708702/article/details/89880098

题目链接
Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

push(x) – Push element x onto stack.
pop() – Removes the element on top of the stack.
top() – Get the top element.
getMin() – Retrieve the minimum element in the stack.

Example:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> Returns -3.
minStack.pop();
minStack.top(); --> Returns 0.
minStack.getMin(); --> Returns -2.

题意:
设计一个最小栈,实现push, pop, top, getMin四个功能。这道题和普通的栈多了一个功能,可以返回当前栈内的最小值。
题解:
使用两个栈,栈1记录进来的数,栈2记录目前的最小值。当有新数push进来的时候,如果栈2为空或者这个数小于栈2顶上的值,就把这个数推入栈2。当pop的数正好等于最小值时,说明当前栈内的最小值变化了,要弹出这个最小值,记录的下一个最小值来到栈顶。

class MinStack {
public:
    /** initialize your data structure here. */
    stack <int> stackValue;
    stack <int> stackMin;
    MinStack() {
        
    }
    
    void push(int x) {
        stackValue.push(x);
        if(stackMin.empty() || x <= stackMin.top()) 
            stackMin.push(x);
    }
    
    void pop() {
        int x = stackValue.top();
        stackValue.pop();
        if(stackMin.top() == x)
            stackMin.pop();
    }
    
    int top() {
        return stackValue.top();
    }
    
    int getMin() {
        return stackMin.top();
    }
};

猜你喜欢

转载自blog.csdn.net/qq_37708702/article/details/89880098
今日推荐