剑指offer30.包含min函数的栈
程序员文章站
2022-07-10 13:58:09
...
Q:定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))
因为时间复杂度要求维O(1),就需要直接读取到该栈中的最小值,考虑到在一个栈上不可能实现随机存取,就需要考虑使用辅助栈记录来完成。
public class Solution {
Stack<Integer> stack1 = new Stack<>();
Stack<Integer> stack2 = new Stack<>();
public void push(int node) {
stack1.push(node);
if (stack2.isEmpty()){
stack2.push(node);
}else {
if (stack2.peek()<node){
stack2.push(stack2.peek());
}else {
stack2.push(node);
}
}
}
public void pop() {
stack1.pop();
stack2.pop();
}
public int top() {
return stack1.peek();
}
public int min() {
return stack2.peek();
}
}
上一篇: python运算符