min函数 辅助栈 c 语言,2021-05-13 包含min函数的栈（辅助栈）

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

示例:

MinStack minStack = new MinStack();

minStack.push(-2);

minStack.push(0);

minStack.push(-3);

minStack.min(); --> 返回 -3.

minStack.pop();

minStack.top(); --> 返回 0.

minStack.min(); --> 返回 -2.

提示：

各函数的调用总次数不超过 20000 次

思路：

两个栈，一个为数据栈，存储所有元素，实现push,pop,top方法；另一个栈为辅助栈，存储A中所有 非严格降序 的元素，实现min方法。

代码：

class MinStack {

// 数据栈，存储所有元素，实现push,pop,top方法

Stack A;

// 辅助栈，存储A中所有 非严格降序 的元素，实现min方法

Stack B;

public MinStack() {

A = new Stack<>();

B = new Stack<>();

}

public void push(int x) {

A.push(x);

// 这里要是>=，避免重复最小值被弹出

if (B.empty() || B.peek() >= x) B.push(x);

}

public void pop() {

// Java 代码中，由于 Stack 中存储的是 int 的包装类 Integer ，因此需要使用 equals() 代替 == 来比较值是否相等。

if (A.pop().equals(B.peek())) B.pop();

}

public int top() {

return A.peek();

}

public int min() {

return B.peek();

}

}