設計一個有getMin功能的棧(2)

阿里瓜瓜發表於2017-05-02
題目：

實現一個特殊的棧，在實現棧的基本功能的基礎上，再實現返回棧中最小元素的操作。

要求：

1、pop、push、getMin操作的時間複雜度都是O(1)
2、設計的棧型別可以輸用現成的棧結構

解答：

藉助另一個棧，記錄stackData棧中的最小元素，【stackMin 棧儲存最小元素】

2

import java.util.Stack;

public class Problem001_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的棧頂元素與getMin()
    * 1. 如果stackMin棧為空，將【newNum】push到stackMin
    * 2. 如果stackMin棧不為空，newNum < 【getMin()】, 將newNum push 至 stackMin
                             value = 【stackMin的棧頂元素】, push value 至 stackMin
    */
    public void push (int newNum){
        if (this.stackData.isEmpty()){
            this.stackMin.push(newNum);
        }else if (newNum <this.getMin()){
            this.stackMin.push(newNum);
        }else {
            int value = this.stackMin.peek();
            this.stackMin.push(value);
        }
        this.stackData.push(newNum);
    }
    
    // pop
       /*  策略： 
        * 1. 如果stackData棧為空，Exception.
        * 2. 如果stackData棧不為空， 【stackData】【stackMin】pop 操作
                                
        */
    public int pop (){
        if(this.stackData.isEmpty()){
            throw new  RuntimeException("Stack is empty");            
        } 
    
        this.stackMin.pop();
        return this.stackData.pop();
    }
    
    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

相關文章