包含min函数的栈

题目描述:

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的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
class
StackWithMin {

Stack<Integer> m_data=
new
Stack<Integer>();
Stack<Integer> m_min=
new
Stack<Integer>();
public
void
push(
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
void
pop() { m_data.pop(); m_min.pop(); }
public
int
top() {
return m_data.peek(); }
public
int
min() {
return m_min.peek(); } }


点赞