包含min函数的栈

来源:互联网 发布:java杨辉三角四行 编辑:程序博客网 时间:2024/06/08 08:43

题目描述:

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。在该栈中,调用min,push及pop的时间复杂度都是O(1).


分析:对于这个题目,需要借助一个成员变量或一个栈来帮助解决问题,但一个成员变量显然是不够的,这里就用到另外一个辅助栈来存储每次数据栈中的最小值。

举个例子:

我们分别压入 5,7,4,2,那么对应的最小值是5,5,4,2,即辅助栈此时为5,5,4,2;

弹出的时候,(5,7,4,2)弹出2,最小值为4,辅助栈也弹出2,为(5,5,4),栈顶4,为最小值;

继续弹出4,数据栈为(5,7),最小值为5,辅助栈为(5,5),栈顶为5,。以此类推。

由此我们可以给出代码:


import java.util.Stack;

public classStackWithMin {

Stack<Integer> m_data=newStack<Integer>();
Stack<Integer> m_min=newStack<Integer>();
public voidpush(int node) {
m_data.push(node);
if(m_min.empty() || node<m_min.peek())
m_min.push(node);
else
m_min.push(m_min.peek());
}
public voidpop() {
m_data.pop();
m_min.pop();
}
public inttop() {
return m_data.peek();
}
public intmin() {
return m_min.peek();
}
}