牛客题霸NC90设计getMin功能的栈Java题解

牛客题霸NC90设计getMin功能的栈Java题解
https://www.nowcoder.com/practice/c623426af02d4c189f92f2a99647bd34?tpId=117&&tqId=35012&rp=1&ru=/ta/job-code-high&qru=/ta/job-code-high/question-ranking
方法：利用辅助栈
解题思路：利用两个栈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();
    }
}