设计一个有getMin功能的栈(1)

题目：

实现一个特殊的栈，在实现栈的基本功能的基础上，再实现返回栈中最小元素的操作。

要求：

1、pop、push、getMin操作的时间复杂度都是O(1)
2、设计的栈类型可以输用现成的栈结构

解答：

借助另一个栈，记录stackData栈中的最小元素，【stackMin 栈保存最小元素】

import java.util.Stack;

public class Problem01_GetMinStack {
    public static class myStack{
    private Stack<Integer> stackData;
    private Stack<Integer> stackMin;
    
    public myStack(){
        this.stackData = new Stack<Integer>();
        this.stackMin = new Stack<Integer>();
    }
    
    
    // push
   /*  策略： 将stackData的栈顶元素与stackMin中栈顶元素做对比
    * 1. 如果stackMin栈为空，将【newNum】push到stackMin
    * 2. 如果stackMin栈不为空，newNum <= 【stackData的栈顶元素】, 将newNum push 至 stackMin
                             newNum > 【stackData的栈顶元素】, 不进行任何操作
    */
    public void push (int newNum){
        if (this.stackData.isEmpty()){
            this.stackMin.push(newNum);
        }else if (newNum <= stackData.peek()){
            this.stackMin.push(newNum);
        }
        this.stackData.push(newNum);
    }
    
    // pop
       /*  策略： 将stackData的栈顶元素与stackMin中栈顶元素做对比
        * 1. 如果stackData栈为空，Exception.
        * 2. 如果stackData栈不为空， 【stackData的栈顶元素】== getMin(), 【stackMin】pop 操作
                                
        */
    public int pop (){
        if(this.stackData.isEmpty()){
            throw new  RuntimeException("Stack is empty");            
        } 
    
        int value = this.stackData.pop();
        if (value == this.getMin()){
            this.stackMin.pop();
        }
        
        return value;
    }
    
    public int getMin() {
        
        if (this.stackMin.isEmpty()){
            throw new RuntimeException("Stack is empty");
            
        }
        return this.stackMin.peek();
        
    }
    }
    public static void main(String[] args) {
        myStack stack1 = new myStack();
        stack1.push(3);
        System.out.println(stack1.getMin());
        stack1.push(4);
        System.out.println(stack1.getMin());
        stack1.push(1);
        System.out.println(stack1.getMin());
        System.out.println(stack1.pop());
        System.out.println(stack1.getMin());
    }

}

执行结果：

3
3
1
1
3