Min Stack 的总结

2024年2月13日 81次阅读

原题地址http://www.lintcode.com/en/problem/min-stack/
一.目的：实现一个最小栈的功能，加入一个新的方法min，使其min的时候返回当前栈中最小值
二.思路：使用两个栈，一个正常栈，一个保存最小值的栈，当调用min方法时候，minstack.peek即可
三.易错点：当重写pop方法时候，正常栈无论如何都要pop，而最小栈只有当栈顶元素与正常栈的栈顶元素相同时候才pop，此处的相等应该用equals方法而不是 == ，否则会出现没有pop出最小栈的栈顶元素的情况；push方法，仅当最小栈为空或者push入的number小于等于（等于不可省略）时才push。

代码如下：

public class MinStack {

    private Stack<Integer> stack;
    private Stack<Integer> minstack;

    public MinStack() {
        stack = new Stack<Integer>();
        minstack = new Stack<Integer>();
    }

    public void push(int number) {
        stack.push(number);
        if(minstack.isEmpty() || number <= minstack.peek()){
            minstack.push(number);
        }
    }

    public int pop() {
        if(stack.peek().equals(minstack.peek())){
            minstack.pop();
        }
        return stack.pop();
    }

    public int min() {
        return minstack.peek();
    }
}
点赞