JAVA 算法 面试题

面试题来源于力扣上的一道面试题

请设计一个栈，除了常规栈支持的pop与push函数以外，还支持min函数，该函数返回栈元素中的最小值。执行push、pop和min操作的时间复杂度必须为O(1)。
示例：
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

JAVA 代码块

class MinStack {
  private Stack<Integer> mainstack;
  private Stack<Integer> minstack;
    /** initialize your data structure here. */
    public MinStack() {
    mainstack=new Stack<Integer>();
    minstack=new Stack<Integer>();
    }
    
    public void push(int x) {
           if(mainstack.empty()){
               mainstack.push(x);
               minstack.push(x);
           } else{
               mainstack.push(x);
               if(minstack.peek()<x){
                   minstack.push(minstack.peek());
               }else {
                   minstack.push(x);
               }
           }
    }
    
    public void pop() {
             mainstack.pop();
             minstack.pop();
    }
    
    public int top() {
             return mainstack.peek();
    }
    
    public int getMin() {
            return minstack.peek();
    }
}

思路就是利用辅助栈 来入出栈 进而达到效果 。
这是我的第一个博客写的不好多多指教