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