剑指 Offer 30. 包含min函数的栈
题目描述:
题解分析:
- 题目其实考察的是栈的知识,本题的目的是使用两个栈来求解最小值。
- 第二个栈主要用来维护第一个栈中的最小值,所以它里面的值都是有序的,值最小的放在栈顶。
- 此外,第二个栈出栈的处理有点难以理解,它是只有当栈顶元素和第一个栈的栈顶元素相同时才会出栈一个元素。
package com.walegarrett.offer;
import java.util.Stack;
/**
* @Author WaleGarrett
* @Date 2021/1/31 13:31
*/
public class Offer_30 {
private Stack<Integer> sta;
private Stack<Integer> minSta;
/** initialize your data structure here. */
public Offer_30() {
sta = new Stack<>();
minSta = new Stack<>();
}
public void push(int x) {
sta.push(x);
if(minSta.isEmpty()){
minSta.push(x);
}else{
int topb = minSta.peek();
if(topb >= x)
minSta.push(x);
}
}
public void pop() {
//这里需要使用equals来判断,否则使用==的话会结果错误
if(sta.pop().equals(minSta.peek()))
minSta.pop();
}
public int top() {
return sta.peek();
}
public int min() {
return minSta.peek();
}
}