题目描述:定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
思路:创建两个栈stack1和stack2,stack2为取得最小值的辅助栈,stack1正常入栈,出栈,取栈顶元素。当stack2为空或者stack2栈顶元素大于等于当前要入栈的元素时,stack2入栈;当stack1出栈时,先判断stack1栈顶元素是否和stack2栈顶元素相同,如果相同则stack2和stack1一同出栈,如果不同则stack1出栈;取栈顶元素返回stack1栈顶元素;取最小值元素返回stack2栈顶元素。
代码:
import java.util.Stack;
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.peek() >= node){
stack2.push(node);
}
}
public void pop() {
if(stack2.peek() == stack1.peek()){
stack2.pop();
}
stack1.pop();
}
public int top() {
return stack1.peek();
}
public int min() {
return stack2.peek();
}
}