包含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();
}
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章