20、包含min函数的栈

a_b_c104

于 2018-01-07 20:32:31 发布

阅读量125

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer

本文链接：https://blog.youkuaiyun.com/a_b_c104/article/details/78997200

剑指offer 专栏收录该内容

65 篇文章

订阅专栏

本文介绍了一种特殊栈的设计方案，通过维护两个栈来高效地获取栈内最小元素。主栈存储所有元素，辅助栈则始终保持其顶部为当前最小值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

思路：构建两个辅助栈，栈1作为数据栈，存储数据；栈2存储当前最小值。push时，保证栈2为最小值。pop时，如果出栈和栈2的栈顶元素相等，栈1栈2都要出栈，保证有效的最小值。

public class Solution {
    Stack<Integer> stack1 = new Stack();
    Stack<Integer> stack2 = new Stack();
    public void push(int node) {
        stack1.push(node);
        if (stack2.empty()) {
            stack2.push(node);//栈2如果为null则直接压入
        } else if (node < stack2.peek()) {//否则如果比栈顶元素小才压入
            stack2.push(node);
        }
    }
    public void pop() {
        if (!stack1.empty()) {
            if (stack1.peek() == stack2.peek()) {
                //如果当前要出栈的元素为当前最小值，栈2也要出栈
                stack2.pop();
            }
            stack1.pop();
        }
    }
    public int top() {
        return stack1.peek();//栈1中的元素为栈顶元素
    }
    public int min() {
        return stack2.peek();
    }
}