【java】剑指offer30_包含min函数的栈

最新推荐文章于 2024-09-16 18:31:59 发布

ic_xcc

最新推荐文章于 2024-09-16 18:31:59 发布

阅读量77

点赞数

分类专栏：算法 java 文章标签：数据结构 java 算法 leetcode 栈

本文链接：https://blog.csdn.net/ic_xcc/article/details/113556134

版权

算法同时被 2 个专栏收录

128 篇文章 0 订阅

订阅专栏

java

122 篇文章 0 订阅

订阅专栏

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.

提示：

各函数的调用总次数不超过 20000 次

参考解题思路：

题目难点--普通栈的 push() 和 pop() 函数的复杂度为 O(1)；而获取栈最小值 min() 函数需要遍历整个栈，复杂度为 O(N)。将 min() 函数复杂度降为 O(1) ，可通过建立辅助栈实现

数据栈 A： 栈 A用于存储所有元素，保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常逻辑。
辅助栈 B ：栈 B中存储栈 A中所有非严格降序的元素，则栈 A中的最小元素始终对应栈 B的栈顶元素，即 min() 函数只需返回栈 B的栈顶元素即可。

因此，只需设法维护好栈 B的元素，使其保持非严格降序，即可实现 min() 函数的 O(1)复杂度。

函数设计：

push(x) 函数：重点为保持栈 B 的元素是非严格降序的。

将 x 压入栈 A（即 A.add(x) ）；
若 ① 栈 B为空或 ② x小于等于栈 B的栈顶元素，则将 x 压入栈 B（即 B.add(x) ）。
pop() 函数：重点为保持栈 A, B的元素一致性。

执行栈 A出栈（即 A.pop() ），将出栈元素记为 y；
若 y 等于栈 B 的栈顶元素，则执行栈 B 出栈（即 B.pop() ）。
top() 函数：直接返回栈 A 的栈顶元素即可，即返回 A.peek() 。

min() 函数：直接返回栈 B的栈顶元素即可，即返回 B.peek() 。

Java 代码中，由于 Stack 中存储的是 int 的包装类 Integer ，因此需要使用 equals() 代替 == 来比较值是否相等。
鉴于pop、top、min在栈为空时可能内存溢出所以首先判断A栈是否为空（因为如果A栈不为空则B栈一定不为空，所以不用判断B）

public class Offer30 {
    Stack<Integer> A,B;

    public MinStack() {
        A = new Stack<Integer>(); //主栈
        B = new Stack<Integer>(); //辅助栈记录最小值保证O(1)
    }

    public void push(int x) {
        A.add(x); //A栈添加元素
        // B为空或者B的栈顶元素大于等于x--则添加进B栈中
        if (B.isEmpty() || B.peek() >= x) {
            B.push(x); //B栈添加元素--目前A栈种的最小元素
        }
    }

    public void pop() {
        // 判断A栈是否为空，避免抛出异常
        if (A.isEmpty()) {
           return;
        }
        if (A.peek().equals(B.peek())) {
            // A栈顶元素与B的栈顶元素相同--移除B栈顶维护的最小值
            B.pop();
        }
        A.pop(); // A栈移出栈顶元素

    }

    public int top() {
        if (A.isEmpty()) {
            return  0; // 返回A的栈顶元素值
        }
        return A.peek(); // 返回A的栈顶元素值
    }

    public int min() {
        if (B.isEmpty()) {
            return 0;
        }
       return B.peek();// 返回B的栈顶元素--因其维护的A栈目前最小值
    }
}