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

最新推荐文章于 2023-07-24 10:28:56 发布

原创最新推荐文章于 2023-07-24 10:28:56 发布 · 944 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种使用辅助栈实现栈数据结构的方法，可以在每次调用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();
    }
}