如何获取栈中最小的数?
程序员文章站
2022-05-12 10:34:08
...
时间限制:1秒 空间限制:32768K 热度指数:171449
算法知识视频讲解
题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路:这道题的难点在于如何在push和pop后更新最小值,比如说我push 3,2,4,0,5,当push3时,stack里最小的元素是3 ;push 2 update min=2 ;push 4 min依然是2 ;push 0 ,update min=0;push 5 min=0;pop min依然是0;pop min由0变为2。在这里,再定义一个存放最小值的栈smin,每push一个值后,相应的push对应的最小值进入smin,pop一个元素,也要pop smin中的一个元素。
package offer;
import java.util.Stack;
public class MinStack {
public Stack<Integer> stack=new Stack<>();
public Stack<Integer> smin=new Stack<>();
public void push(int node) throws Exception{
if(stack.isEmpty()){
stack.push(node);
smin.push(node);
}
else{
if(node<this.min()){
stack.push(node);
smin.push(node);
}
else{
stack.push(node);
smin.push(this.min());
}
}
}
public void pop() {
if(stack.isEmpty()){
return;
}
stack.pop();
smin.pop();
}
public int top() throws Exception{
return (int)stack.peek();
}
public int min() throws Exception{
return (int)smin.peek();
}
public static void main(String[] args) throws Exception{
// TODO Auto-generated method stub
MinStack s=new MinStack();
s.push(2);
s.push(1);
s.push(4);
s.push(0);
while(!s.stack.isEmpty()){
System.out.println(s.min());
s.pop();
}
}
}