常见的思路可能会有:
(1)每入栈一个元素,对栈内元素进行排序,让最小的元素位于栈顶,但是这样不能保证栈的先进后出的特性,不符合题意
(2)设置一个成员变量始终代表最小的元素,每入栈一个元素,若比最小元素小,则更新。但当当前栈顶元素为最小的元素,其出栈后,栈中剩下元素的最小值则无法确定。根据常见思路(2),可以进一步考虑,如果当某个元素入栈时,都能找出当前栈内元素的最小值存入另外一个栈(存放最小值的栈)中,那么当栈顶元素出栈时,
存放最小元素的栈也跟着出栈,那么此时存放最小元素的栈的栈顶元素即为当前栈内最小的元素。
import java.util.Stack;
public class Solution {
//栈1用于存放数据
private Stack<Integer> stack1 = new Stack<>();
//栈2用于存放当前数据栈内的最小元素
private Stack<Integer> stack2 = new Stack<>();
public void push(int node) {
//将数据压入数据栈中
stack1.push(node);
if (stack2.isEmpty()) {
//如果stack2中没有元素,则当前元素即为最小元素,压入stack2中
stack2.push(node);
} else {
//如果stack2中存在元素,那么将当前元素与stack2栈顶元素比较,若比栈顶元素小,则将当前元素入栈,说明当前元素为stack1中最小的元素
//若比stack2栈顶元素大或相等,则将stack2的栈顶元素再次入栈,作为当前stack1中的最小元素
if (node < stack2.peek()) {
stack2.push(node);
} else {
Integer peek = stack2.peek();
stack2.push(peek);
}
}
}
public void pop() {
if (!stack1.isEmpty() && !stack2.isEmpty()) {
//stack1出栈的同时,stack2也必须同步出栈
stack1.pop();
stack2.pop();
}
}
public int top() {
Integer top = Integer.MIN_VALUE;
if (!stack1.isEmpty()) {
top = stack1.peek();
}
return top;
}
public int min() {
if (!stack2.isEmpty()) {
return stack2.peek();
}
return -Integer.MIN_VALUE;
}
}