package java_jianzhioffer_algorithm;
/**
* 题目:定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数
* (时间复杂度应为O(1))。
* @author hexiaoli
* 思考:使用两个stack,一个数据栈存储各个元素,一个辅助栈minStack用来存储最小元素。
* 入栈:stack数据栈入栈,如果minStack非空,比较minStack栈顶元素与入栈stack元素的大小,
* 如果minStack栈顶小,则再入一遍minStack栈顶元素;如果入栈元素小,则将该元素入栈minStack。
* 出栈:数据栈stack和辅助栈minStack均出栈。
* 栈顶元素:直接返回数据栈stack的栈顶元素。
* 栈最小元素:直接返回辅助栈minStack的栈顶元素。
*/
import java.util.Stack;
public class MinFuction {
private static Stack<Integer> stack = new Stack<>();
// 定义一个辅助栈用来存储最小元素
private static Stack<Integer> minStack = new Stack<>();
public static void push(int node) {
// stack入栈
stack.push(node);
// 如果minStack为空,则将该元素入栈
if(minStack.isEmpty()) {
minStack.push(node);
}
// 如果minStack非空,比较minStack栈顶与入栈stack元素的大小,
// 如果minStack栈顶小,则再入一遍栈顶元素;如果入栈元素小,则将该元素入栈minStack
if(minStack.peek()>node) {
minStack.push(node);
}else {
minStack.push(minStack.peek());
}
}
public void pop() {
stack.pop();
minStack.pop();
}
public static int top() {
return stack.peek();
}
public static int min() {
return minStack.peek();
}
public static void main(String[] args) {
MinFuction mFuction = new MinFuction();
stack = mFuction.stack;
minStack = mFuction.minStack;
mFuction.push(5);
mFuction.push(-1);
mFuction.push(4);
int i =mFuction.min();
System.out.println(i);
}
}