题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。
在该栈中调用 min、push、pop的时间复杂度都为O(1)。
思路:
入栈:利用辅助栈,每次入栈时,如果入栈的元素大于辅助栈的栈顶元素,则辅助栈中压入与栈顶相同的元素,如果入栈的元素小于辅助栈栈顶的元素,则把该元素压入辅助栈中。这样辅助栈的栈顶元素就是栈中最小的元素。
出栈:辅助栈和被辅助栈中元素个数始终保持相等,每次出栈操作,辅助栈和被辅助栈都出栈一个元素。
min函数:返回辅助元素的栈顶元素,也就是被辅助栈当前的最小元素。
import java.util.Stack;
public class Min
{
Stack<Integer> data = new Stack<Integer>();//正常存放入栈元素的栈
Stack<Integer> min = new Stack<Integer>();//存放data栈当前最小元素的栈
public void push(int node)//入栈
{
data.push(node);//入栈时正常往data栈中执行压入操作就行,min栈要进行判断后决定压入什么元素
if(min.empty() || min.peek() > node)//如果min为空或者要压入的元素比min栈栈顶元素小,将node元素压入栈中
{
min.push(node);
}
else //如果要压入的元素大于当前min的栈顶元素,则把压入栈顶元素
{
min.push(min.peek());
}
}
public void pop()
{
if(!data.empty())
{
data.pop();
min.pop();
}
}
public int top()
{
return data.peek();
}
public int min()
{
return min.peek();
}
}