1. 程式人生 > 實用技巧 >155. 最小棧(棧)

155. 最小棧(棧)

1、題目描述:

連結:https://leetcode-cn.com/problems/min-stack

設計一個支援 push ,pop ,top 操作,並能在常數時間內檢索到最小元素的棧。

push(x) —— 將元素 x 推入棧中。
pop()—— 刪除棧頂的元素。
top()—— 獲取棧頂元素。
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.

提示:

pop、top 和 getMin 操作總是在 非空棧 上呼叫。

2、思路:

定義兩個棧,一個存所有元素,一個存插入時棧中的最小元素。

劍指offer原題

3、程式碼:

/* 用來存放正常資料 */
    private Stack<Integer> stack;
    /* 用來存放最小資料 */
    private Stack<Integer> minStack;

    /**
     * initialize your data structure here.
     */
    public MinStack() {
        stack = new Stack<>();
        minStack = new Stack<>();
    }

    
public void push(int x) { stack.push(x); if (minStack.isEmpty()) { minStack.push(x); } else { minStack.push(Math.min(x, minStack.peek())); } } public void pop() { stack.pop(); minStack.pop(); } public int top() { return stack.peek(); } public int getMin() { return minStack.peek(); }

。。。