牛客题霸NC90设计getMin功能的栈Java题解
方法:利用辅助栈
解题思路:利用两个栈stackA和stackB,stackA中存储所有元素并负责入栈push()和出栈pop()操作。stackB中存储stackA中所有非严格降序的元素,stackA中最小的元素对应stackB中的栈顶元素。
import java.util.*;
public class Solution {
/**
* return a array which include all ans for op3
* @param op int整型二维数组 operator
* @return int整型一维数组
*/
Stack<Integer> stackA = new Stack<>();
Stack<Integer> stackB = new Stack<>(); //用于存储栈A中所有非严格降序的元素
ArrayList<Integer> tmp = new ArrayList<>(); //用于存储栈中的最小值
public int[] getMinStack (int[][] op) {
for(int[] option : op){
if(option[0]==1){ //如果当前是push操作
push(option[1]);
}else if(option[0]==2){ //如果当前是pop()操作
pop();
}else{ //如果当前是getMin()操作
int min = getMin();
tmp.add(min);
}
}
int[] res = new int[tmp.size()];
for(int i = 0;i< res.length;i++){ //将ArrayList中的元素转存到res数组中,返回
res[i] = tmp.get(i);
}
return res;
}
public void push(int x) {
stackA.add(x); //将x压入stackA中
if(stackB.isEmpty()){ //如果stackB为空,直接压入stackB中
stackB.add(x);
}else{ //如果stackB不为空,且x<=stackB栈顶的元素,则将x压入stackB中
if(x<=stackB.peek()){
stackB.add(x);
}
}
}
public void pop() {
int i = stackA.pop(); //将stackA栈顶的元素弹出
if(i==stackB.peek()){ //如果stackA中弹出的栈顶元素等于stackB中栈顶的元素,则弹出stackB中栈顶的元素
stackB.pop();
}
}
public int getMin() { //stackB中存储stackA中都有非严格降序的元素,所以stackA中最小元素对应stackB的栈顶元素。
return stackB.peek();
}
}