【剑指offer】面试题30：包含min函数的栈

完整代码地址

完整代码地址

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

思路

用一个值来保存最小值是行不通的，因为栈pop操作后，最小值就变了。

但是可以使用另外一个栈minStack来保存最小值。

每次push操作时，将当前最小值与新元素对比得出新的最小值，同时push进minStack。

每次pop操作时，两个栈同时pop。

代码

public class _30_MinInStack {

private Stack<Integer> stack = new Stack<>();
private Stack<Integer> minStack = new Stack<>();

public void push(int node) {
stack.push(node);
if(minStack.isEmpty()) {
minStack.push(node);
}
else if(minStack.peek() > node) {
minStack.push(node);
}
else {
minStack.push(minStack.peek());
}
}

public void pop() {
stack.pop();
minStack.pop();
}

public int top() {
return stack.peek();
}

public int min() {
return minStack.peek();
}
}

测试

public class _30_Test {

public static void main(String[] args) {
test1();
}

/**
* 1.新压入栈的数字比之前的最小值大
* 2.新压入栈的数字比之前的最小值小
* 3.弹出栈的数字不是最小元素
* 4.弹出栈的数字是最小元素
*/
private static void test1() {
_30_MinInStack mis = new _30_MinInStack();
mis.push(4);
mis.push(5);
mis.push(6);
MyTest.equal(mis.top(), 6);
MyTest.equal(mis.min(), 4);

mis.pop();
mis.pop();
MyTest.equal(mis.top(), 4);
MyTest.equal(mis.min(), 4);

mis.push(2);
MyTest.equal(mis.top(), 2);
MyTest.equal(mis.min(), 2);

mis.pop();
MyTest.equal(mis.top(), 4);
MyTest.equal(mis.min(), 4);
}
}