劍指offer 包含min函數的棧

題目描述:
定義棧的數據結構,請在該類型中實現一個能夠得到棧中所含最小元素的min函數。

思路: 定義新數據結構的思路通常都是基於已有數據結構做改造。除了原始的棧,此處需要再使用一個輔助棧和輔助變量,用來存儲當前棧裏面的最小值,在每次往原始棧push值的時候,同時也要更新一個min值,用來添加到min棧裏面。反之,原始棧棧pop的時候也要更新一下min值和min棧

參考代碼:


import java.util.Stack;
public class StackWithMinValue {
    Stack<Integer> SIn = new Stack<Integer>();
    Stack<Integer> SMin = new Stack<Integer>();
    int min = 0;
    public void push(int node) {

        if (SIn.isEmpty()) {
            SIn.push(node);
            SMin.push(node);
            min = node;
        }else if (node > min) {
            SIn.push(node);
            SMin.push(min);
        }else {
            SIn.push(node);
            SMin.push(node);
            min = node;
        }
    }

    public void pop() {
        SMin.pop();
        SIn.pop();
        min = SMin.peek();
    }

    public int top() {

        return SIn.peek();
    }

    public int min() {
        return SMin.peek();
    }

    public static void main(String[] args) {
        StackWithMinValue ss = new StackWithMinValue();
        ss.push(2);
        ss.push(5);
        ss.push(3);
        ss.push(9);
        ss.push(1);

        System.out.println(ss.min());
        System.out.println(ss.top());
        ss.pop();
        System.out.println(ss.min());
        System.out.println(ss.top());
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章