题解 | 包含min函数的栈
包含min函数的栈
https://www.nowcoder.com/practice/4c776177d2c04c2494f2555c9fcc1e49
import java.util.*;
import java.util.Stack;
public class Solution {
Stack<Integer> s1 = new Stack<>();
Stack<Integer> s2 = new Stack<>();
public void push(int node) {
s1.push(node);
// 这里的重点是s2.peek() >= node,需要将等于当前最小值的值也加入到s2中,这样可以确保与s1中与最小值相等的元素数量一致。
if(s2.isEmpty() || s2.peek() >= node){
s2.push(node);
}
}
public void pop() {
int num = s1.pop();
if(num == s2.peek()){
s2.pop();
}
}
public int top() {
return s1.peek();
}
public int min() {
return s2.peek();
}
}
OPPO公司福利 1175人发布