本文已参与「新人创作礼」活动,一起开启掘金创作之路。
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();
*/