题目大意
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
分析
这个题的关键在于要设置两个栈,一个数据栈s_data,一个辅助栈s_min。
- 当进行入栈操作时,如果压入的元素p小于当前最小元素, 辅助栈就压入p,数据栈也压入。 如果压入的元素p大于当前最小元素,辅助栈压入当前最小元素。
- 当进行出栈操作时,辅助栈和数据栈同时弹出元素。
- 当前最小元素取辅助栈的栈顶即可。(要判断s_min是否为空,否则不能取栈顶)
代码
public class Solution {
Stack<Integer>data_s = new Stack<>();
Stack<Integer>min_s = new Stack<>(); //辅助栈
int min_val = Integer.MAX_VALUE;
public void push(int node) {
data_s.push(node);
if(min_s.size() == 0 || node < min_s.peek()) {
min_s.push(node);
} else
min_s.push(min_s.peek());
}
public void pop() {
assert(data_s.size()>0 && min_s.size()>0);
data_s.pop();
min_s.pop();
}
public int top() {
assert(data_s.size()>0);
return data_s.peek();
}
public int min() {
assert(min_s.size()>0);
return min_s.peek();
}
}