题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
思路
定义两个栈:
一个普通栈(存放所有元素),
一个最小栈(仅存放每次进栈操作后当前 最小的元素,栈顶元素就是当前最小元素)
(1)push:
当前元素都直接进入普通栈,
只有当最小栈为空,或者当前待进栈元素小于最小栈的栈顶时,当前元素才可以进入最小栈(否则只能最小栈的栈顶元素进入最小栈…)
(2)pop:
弹出普通栈和最小栈栈顶元素(一定记得最小栈也要弹出,否则求最小元素时两个栈就不对应了)
(3)top:
直接返回普通栈栈顶
(4)min:
直接返回最小栈栈顶
实现
import java.util.Stack;
//要求实现栈结构,还要用栈实现,真的迷
public class Solution {
//因为要求最小元素,所以需要一个辅助栈:专门存放栈中最小的元素(栈顶保证是当前最小的)
Stack<Integer> stack=new Stack<>();
Stack<Integer> minStack=new Stack<>(); //只存放最小元素的栈
public void push(int node) {
//所有元素正常进入普通栈
stack.push(node);
//只有当最小栈为空,或者当前待进栈元素小于最小栈的栈顶时,当前元素可以进入最小栈
if(minStack.isEmpty() || node<minStack.peek()){
minStack.push(node);
}
else{
minStack.push(minStack.peek());
}
}
public void pop() {
stack.pop();
minStack.pop(); //最小栈也要弹出元素,不然求最小值时就不对应了
}
public int top() {
return stack.peek();
}
public int min() {
return minStack.peek();
}
}