剑指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());
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章