设计一个有GetMin功能的栈
【题目】实现一个特殊的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作
【要求】
1.pop、push、getMin操作时间复杂度都是O(1)
2.设计的栈类型可以使用现成的栈结构
【思路】
使用两个栈,一个栈用来保存当前栈中的元素,期功能和一个正常的栈没有区别,这个栈记为stackData;另一个栈用于保存每一步的最小值,这个栈记为StackMin
压入数据规则
1、假设当前数据位newNum,先将其压入stackData,然后判断stackMin是否为空
2、如果为空,则newNum也压入stackMin
3、如果不为空,则比较newNum和stackMin的栈顶元素哪一个更小
4、如果newNum更小或两者相等,则newNum也压入stackMin;
5、如果StackMin中栈顶元素最小,则StackMin不压入任何内容
弹出数据规则
先在stackDat中弹出栈顶元素,记为value,然后比较当前stackMin的栈顶元素和value那一个更小
public class NiuKe { private Stack<Integer> stackData; private Stack<Integer> stackMin; public NiuKe(){ this.stackData =new Stack<Integer>(); this.stackMin = new Stack<Integer>(); } public void push(int newNum){ if (this.stackMin.isEmpty()){ this.stackMin.push(newNum); }else if (newNum <= this.getmin()){ this.stackMin.push(newNum); } this.stackData.push(newNum); } public int pop(){ if (this.stackData.isEmpty()){ throw new RuntimeException("Your stack is Empty."); } int value = this.stackData.pop(); if (value == this.getmin()){ this.stackMin.pop(); } return value; } public int getmin(){ if (this.stackMin.isEmpty()){ throw new RuntimeException("Your stack is empty."); } return this.stackMin.peek(); } }