题目
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
思路
使用辅助栈存储最小值。定义两个栈:数据栈和辅助栈。数据栈存储真正压入的值,辅助站存储每次压栈的最小值。
每次往数据栈压入数据A,同时也往辅助栈压入数B,B = min(A,辅助栈的栈顶元素)。并且每次弹出数据栈栈顶元素,相应的,辅助栈也弹出栈顶元素。那么每次弹出数据栈元素时,辅助栈的栈顶元素永远是现有数据栈的最小值。
白话来讲就是,每次压入就存当时的最小值,所有的最小值都放到辅助栈里记录起来。每次弹出就是找到当时的真相,真相只有一个,就是辅助栈栈顶元素😂
出栈:
出栈轨迹(上图看懂的可以略过此图):
实现
import java.util.Stack;
public class Solution {
//数据栈
private Stack stack = new Stack();
//辅助栈
private Stack minStack = new Stack();
public static void main(String[] args) {
Solution solution = new Solution();
int[] nums = {3, 4, 7, 2, 3};
solution.push(3);
solution.push(4);
solution.push(7);
System.out.println(solution.min());
solution.pop();
System.out.println(solution.min());
solution.pop();
System.out.println(solution.min());
}
/**
* 插入数据
* @param node
*/
public void push(int node) {
stack.push(node);
minPush(node);
}
/**
* 往辅助栈插入数据
* @param node
*/
public void minPush(int node){
//这个数和辅助栈做比较
int minNum = node;
if (minStack.size() > 0) {
int peek = (Integer)minStack.peek();
minNum = node < peek ? node : peek;
}
minStack.push(minNum);
}
/**
* 弹出数据栈
*/
public void pop() {
stack.pop();
minStack.pop();
}
/**
* 获取数据栈最顶数据
* @return
*/
public int top() {
return (Integer)stack.peek();
}
/**
* 获取栈的最小值
* @return
*/
public int min() {
int peek = (Integer)minStack.peek();
return peek;
}
}