题目描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数。
思路: 定义新数据结构的思路通常都是基于已有数据结构做改造。除了原始的栈,此处需要再使用一个辅助栈和辅助变量,用来存储当前栈里面的最小值,在每次往原始栈push值的时候,同时也要更新一个min值,用来添加到min栈里面。反之,原始栈栈pop的时候也要更新一下min值和min栈
参考代码:
import java.util.Stack;
public class StackWithMinValue {
Stack<Integer> SIn = new Stack<Integer>();
Stack<Integer> SMin = new Stack<Integer>();
int min = 0;
public void push(int node) {
if (SIn.isEmpty()) {
SIn.push(node);
SMin.push(node);
min = node;
}else if (node > min) {
SIn.push(node);
SMin.push(min);
}else {
SIn.push(node);
SMin.push(node);
min = node;
}
}
public void pop() {
SMin.pop();
SIn.pop();
min = SMin.peek();
}
public int top() {
return SIn.peek();
}
public int min() {
return SMin.peek();
}
public static void main(String[] args) {
StackWithMinValue ss = new StackWithMinValue();
ss.push(2);
ss.push(5);
ss.push(3);
ss.push(9);
ss.push(1);
System.out.println(ss.min());
System.out.println(ss.top());
ss.pop();
System.out.println(ss.min());
System.out.println(ss.top());
}
}