剑指offer题21 包含min函数的栈

题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。


题目解析:
题目的重点在于,每次调用min函数都可以pop出当前栈中的最小值;
可以考虑使用一个辅助栈minStack,当每次push进来一个数值node时,minStack将当前栈顶值(默认当前最小值)与node进行比较,push捡来最小值;则在每个时刻,minStack中保存的都是当前栈中的最小值;然后按顺序pop出来就行了;(要注意pop时,数据栈和辅助栈中的数据要保持同步,因此应该同时pop)

代码:
public class Solution {
    private Stack<Integer> dataStack = new Stack<>(); // 数据Stack
    private Stack<Integer> minStack  = new Stack<>(); // 保存最小值的辅助Stack

    public void push(int node) {
        dataStack.push(node);

        if (minStack.isEmpty())
            minStack.push(node);
        else
            minStack.push(Math.min(node, minStack.peek()));
    }

    public void pop() {
        dataStack.pop();
        minStack.pop();
    }

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

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


你可能感兴趣的:(剑指offer题21 包含min函数的栈)