LeetCode-155-最小栈

最小栈

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

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

示例说明请见LeetCode官网。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-stack/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法一:双栈

利用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 data;
    private Stack 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();
    }
}

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

你可能感兴趣的:(LeetCode-155-最小栈)