题目
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
解题思路
第一步:使用一个min变量保存栈中的最小值
第二步:在每次对栈中进行增删时判断最小值是否改变,如果改变则维护min的值。
代码如下
Stack<Integer> stack = new Stack<>();
int min = Integer.MAX_VALUE;
public void push(int node) {
if (node < min) {
min = node;
}
stack.add(node);
}
public void pop() {
Integer pop = stack.pop();
if (pop == min){
min = Integer.MAX_VALUE;
for (int val: stack) {
if (val < min)
min = val;
}
}
}
public int top() {
Integer remove = stack.remove(0);
if (remove == min){
min = Integer.MAX_VALUE;
for (int val: stack) {
if (val < min)
min = val;
}
}
return remove;
}
public int min() {
return min;
}