劍指Offer:面試題21——包含min函數的棧(java實現)

問題描述：

定義棧的數據結構，請在該類型中實現一個能夠得到棧的最小元素的min函數。在該棧中，調用min,push及pop的時間複雜度都是O(1).

思路：加入一個輔助棧用來存儲最小值集合

（這裏要注意題目並沒有說棧內的元素類型，因此要儘量通用）

import java.util.Stack;

public class Solution {


    Stack s1=new Stack();
    Stack min=new Stack();
    public void push(int node) {
        if(min.empty()){
            min.push(node);
        }else{
            int top=(int)min.peek();
            if(node<top){
               min.push(node);
            }else{
                min.push(top);
            }
        }
        s1.push(node);
    }

    public void pop() {
        if(!(s1.empty())){
            s1.pop();
            min.pop();
        }
    }

    public int top() {
        return (int)s1.peek();
    }

    public int min() {
       if(min.empty()){
           return 0;
       }
       return (int)min.peek();    
    }





}