【单调栈】解决含有相同元素的单调栈

最新推荐文章于 2024-07-14 22:26:46 发布

强连通子图

最新推荐文章于 2024-07-14 22:26:46 发布

阅读量1.1k

点赞数 27

文章标签： java 算法单调栈

本文链接：https://blog.csdn.net/weixin_46879314/article/details/136781995

版权

本文介绍了在单调栈中处理含有相同值的情况，通过链表链接下标，探讨了新元素的处理策略和弹出时的更新方法。通过实例和代码详细展示了有重复值单调栈的实现过程，并强调了无重复值特殊情况下的处理区别。

摘要由CSDN通过智能技术生成

上篇文章我们讨论了单调栈的相关知识，并给出了 不含相同值的单调栈 如何实现的代码。（还没看过上篇文章的赶快点我查看哦！）

本文我们继续深入学习，思考 含有相同值的单调栈 如何实现。

由已知推未知：没有相同值的单调栈中我们存放的是各元素的下标。因此，当含有相同值时，它们的 下标是不一样的 ，我们用链表将 相同值 的元素 下标链接起来。

此时单调栈中存放的就是一个一个的链表了。

实现思路

上次我们寻找的是两侧最近且大于某个数的，这次我们换成 寻找最近且小于某个数 的（其实本质都一样，换个“>”、"<"号即可）。（栈底小 -> 栈顶大）：

新元素的处理： 当一个新元素到来时，先与栈顶元素进行比较。

若比栈顶元素大：创建一个新的链表后直接压栈保存；
若比栈顶元素小：需要弹出栈顶链表中所有元素，直到栈空或比栈顶元素大后压栈。
若遇到相同大小时：加入到链表尾部。

弹出时更新信息： 当弹出栈顶元素时，此时次栈顶元素和新来元素就是所求的左右两侧距离最近的且小于该元素的值，更新该链表中所有元素信息 。

没看懂！没关系，看图例：

注意： 栈中存放的是元素的下标。

代码

public static int[][] getNearLess(int[] arr) {
    int n = arr.length;
    int[][] res = new int[n][2];
    // 链表栈
    Stack<List<Integer>> stack = new Stack<>();
    for (int i = 0; i < n; i++) {
        while (!stack.isEmpty() && arr[stack.peek().get(0)] > arr[i]) {
            List<Integer> popls = stack.pop();
            int leftLessIndex = stack.isEmpty() ? -1 : stack.peek().get(stack.peek().size() - 1);
            for (Integer popi : popls) {
                res[popi][0] = leftLessIndex;
                res[popi][1] = i;
            }
        }
        // 相等时加入
        if (!stack.isEmpty() && arr[stack.peek().get(0)] == arr[i]) {
            stack.peek().add(Integer.valueOf(i));
        } else {
        // 小于时，新增一个链表，加入栈顶
            ArrayList<Integer> list = new ArrayList<>();
            list.add(i);
            stack.push(list);
        }
    }
    while (!stack.isEmpty()) {
        List<Integer> popls = stack.pop();
        // 从链表尾部依次弹出
        int leftLessIndex = stack.isEmpty() ? -1 : stack.peek().get(stack.peek().size() - 1);
        for (Integer popi : popls) {
            res[popi][0] = leftLessIndex;
            res[popi][1] = -1;
        }
    }
    return res;
}