力扣456： 132模式（单调栈）

最新推荐文章于 2023-07-09 22:42:52 发布

三水写代码

最新推荐文章于 2023-07-09 22:42:52 发布

阅读量301

点赞数

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

原文链接：https://leetcode-cn.com/problems/132-pattern/solution/132mo-shi-by-leetcode-solution-ye89/

版权

数据结构与算法专栏收录该内容

8 篇文章 0 订阅

订阅专栏

给你一个整数数组 nums ，数组中共有 n 个整数。132 模式的子序列由三个整数 nums[i]、nums[j] 和 nums[k] 组成，并同时满足：i < j < k 和 nums[i] < nums[k] < nums[j] 。

如果 nums 中存在 132 模式的子序列，返回 true ；否则，返回 false 。

进阶：很容易想到时间复杂度为 O(n^2) 的解决方案，你可以设计一个时间复杂度为 O(n logn) 或 O(n) 的解决方案吗？

方法一：枚举 3（最大值）

思路与算法

枚举 3 是容易想到并且也是最容易实现的。由于 3 是模式中的最大值，并且其出现在 1和 2 的中间，因此我们只需要从左到右枚举 3 的下标 j，那么：

由于 1 是模式中的最小值，因此我们在枚举 jjj 的同时，维护数组 aaa 中左侧元素 a[0..j−1]的最小值，即为 1 对应的元素 a[i]]。需要注意的是，只有 a[i]<a[j] 时，a[i]才能作为 1 对应的元素；
由于 2 是模式中的次小值，因此我们可以使用一个有序集合（例如平衡树）维护数组 aaa 中右侧元素 a[j+1..n−1]中的所有值。当我们确定了 a[i] 和 a[j] 之后，只需要在有序集合中查询严格比 a[i]大的那个最小的元素，即为 a[k]。需要注意的是，只有 a[k]<a[j] 时，a[k]才能作为 3 对应的元素。

代码：

class Solution {
    public boolean find132pattern(int[] nums) {
        int n = nums.length;
        if (n < 3) {
            return false;
        }

        // 左侧最小值
        int leftMin = nums[0];
        // 右侧所有元素
        TreeMap<Integer, Integer> rightAll = new TreeMap<Integer, Integer>();

        for (int k = 2; k < n; ++k) {
            rightAll.put(nums[k], rightAll.getOrDefault(nums[k], 0) + 1);
        }

        for (int j = 1; j < n - 1; ++j) {
            if (leftMin < nums[j]) {
                Integer next = rightAll.ceilingKey(leftMin + 1);
                if (next != null && next < nums[j]) {
                    return true;
                }
            }
            leftMin = Math.min(leftMin, nums[j]);
            rightAll.put(nums[j + 1], rightAll.get(nums[j + 1]) - 1);
            if (rightAll.get(nums[j + 1]) == 0) {
                rightAll.remove(nums[j + 1]);
            }
        }

        return false;
    }
}

复杂度分析

时间复杂度：O(nlog⁡n)。在初始化时，我们需要 O(nlog⁡n) 的时间将数组元素 a[2..n−1]加入有序集合中。在枚举 j 时，维护左侧元素最小值的时间复杂度为 O(1)，将 a[j+1]从有序集合中删除的时间复杂度为 O(log⁡n)，总共需要枚举的次数为 O(n)，因此总时间复杂度为 O(nlog⁡n)。
空间复杂度：O(n)，即为有序集合存储右侧所有元素需要使用的空间。

方法二：枚举 1（最小值）

思路与算法：

如果我们从左到右枚举 1 的下标 i，那么 j,k 的下标范围都是减少的，这样就不利于对它们进行维护。因此我们可以考虑从右到左枚举 i。

那么我们应该如何维护 j,k 呢？在 132 模式中，如果 1<2 并且 2<3，那么根据传递性，1<3 也是成立的，那么我们可以使用下面的方法进行维护：

我们使用一种数据结构维护所有遍历过的元素，它们作为 2 的候选元素。每当我们遍历到一个新的元素时，就将其加入数据结构中；

在遍历到一个新的元素的同时，我们可以考虑其是否可以作为 3。如果它作为 3，那么数据结构中所有严格小于它的元素都可以作为 2，我们将这些元素全部从数据结构中移除，并且使用一个变量维护所有被移除的元素的最大值。这些被移除的元素都是可以真正作为 2 的，并且元素的值越大，那么我们之后找到 1 的机会也就越大。

那么这个「数据结构」是什么样的数据结构呢？我们尝试提取出它进行的操作：

它需要支持添加一个元素；
它需要支持移除所有严格小于给定阈值的所有元素；
上面两步操作是「依次进行」的，即我们先用给定的阈值移除元素，再将该阈值加入数据结构中。

这就是「单调栈」。在单调栈中，从栈底到栈顶的元素是严格单调递减的。当给定阈值 x 时，我们只需要不断地弹出栈顶的元素，直到栈为空或者 x 严格小于栈顶元素。此时我们再将 x 入栈，这样就维护了栈的单调性。

因此，我们可以使用单调栈作为维护 2 的数据结构，并给出下面的算法：

我们用单调栈维护所有可以作为 2 的候选元素。初始时，单调栈中只有唯一的元素 a[n−1]。我们还需要使用一个变量 max_k 记录所有可以真正作为 2 的元素的最大值；

随后我们从 n−2 开始从右到左枚举元素 a[i]：

首先我们判断 a[i] 是否可以作为 1。如果 a[i]<max_ka[i] ，那么它就可以作为 1，我们就找到了一组满足 132 模式的三元组；
随后我们判断 a[i] 是否可以作为 3，以此找出哪些可以真正作为 2 的元素。我们将 a[i] 不断地与单调栈栈顶的元素进行比较，如果 a[i] 较大，那么栈顶元素可以真正作为 2，将其弹出并更新 max_k；
最后我们将 a[i] 作为 2 的候选元素放入单调栈中。这里可以进行一个优化，即如果 a[i]≤max_k，那么我们也没有必要将 a[i] 放入栈中，因为即使它在未来被弹出，也不会将 max_k 更新为更大的值。

在枚举完所有的元素后，如果仍未找到满足 132 模式的三元组，那就说明其不存在。

代码

class Solution {
    public boolean find132pattern(int[] nums) {
        int n = nums.length;
        Deque<Integer> candidateK = new LinkedList<Integer>();
        candidateK.push(nums[n - 1]);
        int maxK = Integer.MIN_VALUE;

        for (int i = n - 2; i >= 0; --i) {
            if (nums[i] < maxK) {
                return true;
            }
            while (!candidateK.isEmpty() && nums[i] > candidateK.peek()) {
                maxK = candidateK.pop();
            }
            if (nums[i] > maxK) {
                candidateK.push(nums[i]);
            }
        }

        return false;
    }
}

复杂度分析

时间复杂度：O(n)，枚举 i 的次数为 O(n)，由于每一个元素最多被加入和弹出单调栈各一次，因此操作单调栈的时间复杂度一共为 O(n)，总时间复杂度为 O(n)。
空间复杂度：O(n)，即为单调栈需要使用的空间。

方法三：枚举 2（中间值）

说明：方法三思路难度较大，需要在单调栈上进行二分查找。建议读者在完全理解方法二之后，再尝试阅读该方法。

思路与算法

当我们枚举 222 的下标 kkk 时，与方法二相反，从左到右进行枚举的方法是十分合理的：在枚举的过程中，i,j 的下标范围都是增加的。

由于我们需要保证 1<2 并且 2<3，那么我们需要维护一系列尽可能小的元素作为 1 的候选元素，并且维护一系列尽可能大的元素作为 3 的候选元素。

我们可以分情况进行讨论，假设当前有一个小元素 xix_ixi 以及一个大元素 xj 表示一个二元组，而我们当前遍历到了一个新的元素 x=a[k]，那么：

如果 x>xj，那么让 x 作为 3 显然是比 x作为 3 更优，因此我们可以用 x 替代 xj；

如果 x<xi，那么让 x 作为 1 显然是比 xi 作为 3 更优，然而我们必须要满足 132 模式中的顺序，即 1 出现在 3 之前，这里如果我们简单地用 x 替代 x，那么 xi=x 作为 1 是出现在 xj 作为 3 之后的，这并不满足要求。因此我们需要为 x 找一个新的元素作为 3。由于我们还没有遍历到后面的元素，因此可以简单地将 x 同时看作一个二元组的 xi 和 xj；

对于其它的情况，xi≤x≤xj，x 无论作为 1 还是 3 都没有当前二元组对应的要优，因此我们可以不用考虑 x 作为 1 或者 3 的情况。

这样一来，与方法二类似，我们使用两个单调递减的单调栈维护一系列二元组 (xi,xj)，表示一个可以选择的 1−3 区间，并且从栈底到栈顶 xi 和 x 分别严格单调递减，因为根据上面的讨论，我们只有在 x<xi 时才会增加一个新的二元组。

然而与方法二不同的是，如果我们想让 x 作为 2，那么我们并不知道到底应该选择单调栈中的哪个 1−3 区间，因此我们只能根据单调性进行二分查找：

对于单调栈中的 xi，我们需要找出第一个满足 xi<x 的位置 idxi，这样从该位置到栈顶的所有二元组都满足 xi<x；
对于单调栈中的 xj，我们需要找出最后一个满足 xj>x 的位置 idxj，这样从栈底到该位置的所有二元组都满足 xj>x；
如果 idxi 和 idxj 都存在，并且 idxi≤idxj，那么就存在至少一个二元组 (xi,xj) 满足 xi<x<xj，xxx 就可以作为 2，我们就找到了一组满足 132 模式的三元组。

在枚举完所有的元素后，如果仍未找到满足 132 模式的三元组，那就说明其不存在。

代码

需要注意的是，我们是在单调递减的栈上进行二分查找，因此大部分语言都需要实现一个自定义比较函数，或者将栈中的元素取相反数后再使用默认的比较函数。

class Solution {
    public boolean find132pattern(int[] nums) {
        int n = nums.length;
        List<Integer> candidateI = new ArrayList<Integer>();
        candidateI.add(nums[0]);
        List<Integer> candidateJ = new ArrayList<Integer>();
        candidateJ.add(nums[0]);

        for (int k = 1; k < n; ++k) {
            int idxI = binarySearchFirst(candidateI, nums[k]);
            int idxJ = binarySearchLast(candidateJ, nums[k]);
            if (idxI >= 0 && idxJ >= 0) {
                if (idxI <= idxJ) {
                    return true;
                }
            }
            
            if (nums[k] < candidateI.get(candidateI.size() - 1)) {
                candidateI.add(nums[k]);
                candidateJ.add(nums[k]);
            } else if (nums[k] > candidateJ.get(candidateJ.size() - 1)) {
                int lastI = candidateI.get(candidateI.size() - 1);
                while (!candidateJ.isEmpty() && nums[k] > candidateJ.get(candidateJ.size() - 1)) {
                    candidateI.remove(candidateI.size() - 1);
                    candidateJ.remove(candidateJ.size() - 1);
                }
                candidateI.add(lastI);
                candidateJ.add(nums[k]);
            }
        }

        return false;
    }

    public int binarySearchFirst(List<Integer> candidate, int target) {
        int low = 0, high = candidate.size() - 1;
        if (candidate.get(high) >= target) {
            return -1;
        }
        while (low < high) {
            int mid = (high - low) / 2 + low;
            int num = candidate.get(mid);
            if (num >= target) {
                low = mid + 1;
            } else {
                high = mid;
            }
        }
        return low;
    }

    public int binarySearchLast(List<Integer> candidate, int target) {
        int low = 0, high = candidate.size() - 1;
        if (candidate.get(low) <= target) {
            return -1;
        }
        while (low < high) {
            int mid = (high - low + 1) / 2 + low;
            int num = candidate.get(mid);
            if (num <= target) {
                high = mid - 1;
            } else {
                low = mid;
            }
        }
        return low;
    }
}

复杂度分析

时间复杂度：O(nlog⁡n)，枚举 i 的次数为 O(n)，由于每一个元素最多被加入和弹出单调栈各一次，因此操作单调栈的时间复杂度一共为 O(n)。二分查找的单次时间为 O(log⁡n)，一共为 O(nlog⁡n)，总时间复杂度为 O(nlog⁡n)。
空间复杂度：O(n)，即为单调栈需要使用的空间。

结语：

在上面的三种方法中，方法二的时间复杂度为 O(n)，最优秀。而剩余的两种时间复杂度为 O(nlog⁡n)的方法中，方法一相较于方法三，无论从理解还是代码编写层面来说都更容易一些。那么为什么还要介绍方法三呢？这里我们可以发现方法一和方法二的不足：

方法一需要提前知道整个数组，否则就无法使用有序集合维护右侧元素了；
方法二是从后向前遍历的，本质上也同样需要提前知道整个数组。
方法三是从前向后遍历的，并且维护的数据结构不依赖于后续未知的元素，因此如果数组是以「数据流」的形式给出的，那么方法三是唯一可以继续使用的方法。

链接：https://leetcode-cn.com/problems/132-pattern/solution/132mo-shi-by-leetcode-solution-ye89/
来源：力扣（LeetCode）

三水写代码

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
力扣456： 132模式（单调栈）

给你一个整数数组 nums ，数组中共有 n 个整数。132 模式的子序列由三个整数 nums[i]、nums[j] 和 nums[k] 组成，并同时满足：i < j < k 和 nums[i] < nums[k] < nums[j] 。如果 nums 中存在 132 模式的子序列，返回 true ；否则，返回 false 。进阶：很容易想到时间复杂度为 O(n^2) 的解决方案，你可以设计一个时间复杂度为 O(n logn) 或 O(n) 的解决方案吗？方法..
复制链接

扫一扫

专栏目录