LeetCode-155-最小栈

Posted 2021-09-03 雄狮虎豹

最小栈

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

• push(x) —— 将元素 x 推入栈中。
• pop() —— 删除栈顶的元素。
• top() —— 获取栈顶元素。
• getMin() —— 检索栈中的最小元素。

示例说明请见LeetCode官网。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/probl...
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一：双栈

利用2个栈，一个栈data存储数据，一个栈minVal存储最小值，具体方法逻辑：

• push()：将val值放到data的栈顶，判断val值是否小于等于minVal的栈顶元素，如果是，则将val值也放到minVal的栈顶；
• pop()：将data的栈顶元素取出，判断如果取出的元素等于minVal的栈顶元素，则将minVal的栈顶元素也取出；
• top()：查看data的栈顶元素；
• getMin()：查看minVal的栈顶元素。

import java.util.Stack;

public class LeetCode_155 {
    public static void main(String[] args) {
        MinStack minStack = new MinStack();
        minStack.push(-2);
        minStack.push(0);
        minStack.push(-3);
        System.out.println(minStack.getMin());   // 返回 -3.
        minStack.pop();
        System.out.println(minStack.top());      // 返回 0.
        System.out.println(minStack.getMin());   // 返回 -2.
    }
}

class MinStack {
    private Stack<Integer> data;
    private Stack<Integer> minVal;

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

    public void push(int val) {
        data.push(val);
        if (minVal.isEmpty()) {
            minVal.push(val);
        } else if (val <= minVal.peek()) {
            minVal.push(val);
        }
    }

    public void pop() {
        if (data.isEmpty()) {
            throw new RuntimeException("stack is empty.");
        }
        int popVal = data.pop();
        if (popVal == minVal.peek()) {
            minVal.pop();
        }
    }

    public int top() {
        return data.peek();
    }

    public int getMin() {
        if (minVal.isEmpty()) {
            throw new RuntimeException("stack is empty.");
        }
        return minVal.peek();
    }
}

【每日寄语】 不去追逐，永远不会拥有。不往前走，永远原地停留。知道自己目的地的人，才是旅行得最远的人。