Leetcode 刷题:剑指 Offer 30. 包含min函数的栈,Easy(Java)

177 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

1.题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:

MinStack minStack = new MinStack();

minStack.push(-2);

minStack.push(0);

minStack.push(-3);

minStack.min(); --> 返回 -3.

minStack.pop();

minStack.top(); --> 返回 0.

minStack.min(); --> 返回 -2.

提示:

各函数的调用总次数不超过 20000 次

来源:力扣(LeetCode) 链接:剑指 Offer 30. 包含min函数的栈 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解法

2.1 思路

Easy题我试了三次,还是要注意细节。

tips: 返回值是对象时最好不要用 == ,而是用 equals()

还是讲一下思路,因为有最小值需要管理,所以需要另外一个minStack栈,用一个值是不够的,因为会变,所以这里用一个栈来存储先后出现的最小值。 如果后出现的,按照栈的特性会先出去,刚好满足了最小值的变化。 push的时候判断一下是不是当前的最小值,是的话就放进minStack,其实可以发现minStack的是递减的。虽然没啥用。这个时候就已经完成了最小值的更新。 pop的时候判断一下,此时pop()出来的值,是不是于minStack栈顶的值相同,如果相同说明最小值也要一起pop()。

代码不写注释是因为想看看我的代码好不好懂,不好懂请在评论区喷我,下次就改。

2.2 Java代码

class MinStack {
    Stack<Integer> stack;
    Stack<Integer> minStack;
    /** initialize your data structure here. */
    public MinStack() {
        stack = new Stack<Integer>();
        minStack = new Stack<Integer>();
    }
    
    public void push(int x) {
        stack.push(x);
        if(minStack.empty()||minStack.peek()>=x){
            minStack.push(x);
        }
    }
    
    public void pop() {
        if(stack.peek().equals(minStack.peek())){
            minStack.pop();
        }
        stack.pop();        
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int min() {
        return minStack.peek();
    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.min();
 */