題目描述:
定義棧的數據結構,請在該類型中實現一個能夠得到棧中所含最小元素的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());
}
}