题目描述:设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。 示例:输入: ["MinStack","push","push","push","getMin","pop","top","getMin"] [[],[-2],[0],[-3],[],[],[],[]]
输出: [null,null,null,null,-3,null,0,-2]
解释: MinStack minStack = new MinStack(); minStack.push(-2); minStack.push(0); minStack.push(-3); minStack.getMin(); --> 返回 -3. minStack.pop(); minStack.top(); --> 返回 0. minStack.getMin(); --> 返回 -2.
提示:pop 、 top 和 getMin 操作总是在 非空栈 上调用。
题目分析:这道题最简单的做法就是建立一个最小值栈,栈顶的元素代表基本栈里所有值中的最小值。首先看看压入操作,要往基本栈里压入一个元素,要检查最小值栈是否为空,如果最小值栈为空,证明基本栈也为空,所以要把这个元素一并压入最小值栈。当然如果最小值栈的栈顶元素大于等于要压入的元素时,就要更新最小值栈,把该元素即新的最小值元素压入最小值栈。再来看看弹出操作,如果最小值栈非空且最小值栈栈顶元素与基本栈栈顶元素相同,我们就先要弹出最小值栈栈顶元素,再弹出基本栈中的栈顶元素。最后获取基本栈所有元素中的最小值就是获取最小值栈栈顶元素的值了。 题解:执行用时: 5 ms 内存消耗: 40 MB class MinStack {
// 基本栈 Stack<Integer> stack; // 最小值栈 Stack<Integer> minStack;
public MinStack() { stack = new Stack<>(); minStack = new Stack<>(); }
public void push(int val) { // 把值压入基本栈 stack.push(val); // 如果最小值值为空 或者 最小值栈顶元素大于等于当前值 if (minStack.empty() || minStack.peek() >= val) { // 把当前元素压入最小值栈 minStack.push(val); } }
public void pop() { // 当最小值栈非空 且 最小值栈栈顶元素与基本栈栈顶元素相同 if (!minStack.empty() && minStack.peek().equals(stack.peek())) { // 先弹出最小值栈栈顶元素 minStack.pop(); } // 弹出基本栈栈顶元素 stack.pop(); }
public int top() { // 获取基本栈栈顶元素 return stack.peek(); }
public int getMin() { // 获取基本栈最小元素即为获取最小值栈栈顶元素 return minStack.peek(); } }
/** * Your MinStack object will be instantiated and called as such: * MinStack obj = new MinStack(); * obj.push(val); * obj.pop(); * int param_3 = obj.top(); * int param_4 = obj.getMin(); */
|