单调栈题目：最大宽度坡

题目

标题和出处

标题：最大宽度坡

出处：962. 最大宽度坡

难度

7 级

题目描述

要求

整数数组 $\text{nums}$ 中的坡是数组对 $\text{(i, j)}$，其中 $\text{i}<\text{j}$ 且 $\text{nums[i]}\le \text{nums[j]}$。这样的坡的宽度为 $\text{j - i}$。

给定整数数组 $\text{nums}$，找出 $\text{nums}$ 中的坡的最大宽度。如果 $\text{nums}$ 中不存在坡，返回 $\text{0}$。

示例

示例 1：

输入：$\text{nums = [6,0,8,2,1,5]}$
输出：$\text{4}$
解释：最大宽度的坡为 $\text{(i, j) = (1, 5)}$：$\text{nums[1] = 0}$ 且 $\text{nums[5] = 5}$。

示例 2：

输入：$\text{nums = [9,8,1,0,1,9,4,0,4,1]}$
输出：$\text{7}$
解释：最大宽度的坡为 $\text{(i, j) = (2, 9)}$：$\text{nums[2] = 1}$ 且 $\text{nums[9] = 1}$。

数据范围

• $\text{2}\le \text{nums.length}\le \text{5}\times {\text{10}}^{\text{4}}$
• $\text{0}\le \text{nums[i]}\le \text{5}\times {\text{10}}^{\text{4}}$

解法

思路和算法

为了得到数组 $\text{nums}$ 中的坡的最大宽度，需要对每个下标 $j$ 找到最小的下标 $i$，满足 $i<j$ 且 $\text{nums}[i]\le \text{nums}[j]$。

最简单的做法是对于每个下标 $j$，遍历从 $0$ 到 $j-1$ 的全部下标，找到最小的满足 $\text{nums}[i]\le \text{nums}[j]$ 的下标 $i$。假设数组 $\text{nums}$ 的长度是 $n$，该做法的时间复杂度是 $O(n^2)$，会超出时间限制，因此需要优化。

假设存在一个坡 $(i,j)$，则有 $i<j$ 且 $\text{nums}[i]\le \text{nums}[j]$。如果存在下标 $p$ 和 $q$ 满足 $i<p<j<q$ 且 $\text{nums}[p]\le \text{nums}[q]$，考虑 $\text{nums}[i]$ 和 $\text{nums}[p]$ 的大小关系。

• 如果 $\text{nums}[i]\le \text{nums}[p]$，则即使有 $\text{nums}[p]\le \text{nums}[j]$，以 $j$ 结尾的最大宽度坡（即所有以 $j$ 结尾的坡中宽度最大的坡）也不可能以 $p$ 开始，因为 $j-i>j-p$，以 $i$ 开始的坡宽度更大，对于 $q$ 而言，$\text{nums}[i]$ 和 $\text{nums}[p]$ 都小于等于 $\text{nums}[q]$，由于 $q-i>q-p$，因此以 $q$ 结尾的最大宽度坡不可能以 $p$ 开始。

• 如果 $\text{nums}[i]>\text{nums}[p]$，则虽然以 $j$ 结尾的最大宽度坡不可能以 $p$ 开始，但是由于 $\text{nums}[q]$ 可能小于 $\text{nums}[i]$，因此以 $q$ 结尾的最大宽度坡可能以 $p$ 开始。

根据上述分析可知，下标 $i$ 可能是最大宽度坡的开始下标的条件是对于任意小于 $i$ 的下标 $k$ 都有 $\text{nums}[k]>\text{nums}[i]$。如果存在一个坡 $(i,j)$ 且存在小于 $i$ 的下标 $k$ 满足 $\text{nums}[k]\le \text{nums}[i]$，则 $(k,j)$ 一定是坡，且坡 $(k,j)$ 的宽度比坡 $(i,j)$ 的宽度更大。

如果可以排除不可能是最大宽度坡的开始下标的下标，则可以降低时间复杂度。可以使用单调栈存储可能是最大宽度坡的开始下标的全部下标，单调栈满足从栈底到栈顶的下标对应的元素单调递减。

从左到右遍历数组 $\text{nums}$，对于每个下标 $i$，当且仅当栈为空或者栈顶下标对应的元素大于 $\text{nums}[i]$ 时，将 $i$ 入栈。遍历结束之后，栈内的每个下标 $i$ 都满足对于任意小于 $i$ 的下标 $k$ 都有 $\text{nums}[k]>\text{nums}[i]$。

然后从右到左遍历数组 $\text{nums}$，对于每个下标 $j$，需要找到最小的下标 $i$ 使得 $\text{nums}[i]\le \text{nums}[j]$。具体做法是，当栈不为空且栈顶下标对应的元素小于等于 $\text{nums}[j]$ 时，令栈顶下标为 $i$，将 $i$ 出栈，并用 $j-i$ 更新坡的最大宽度，重复该操作直到栈为空或者栈顶下标对应的元素大于 $\text{nums}[j]$。该做法的正确性说明如下。

1. 对于下标 $j$，如果有多个小于 $j$ 的下标对应的元素都小于等于 $\text{nums}[j]$，则其中最小的下标和 $j$ 组成以 $j$ 结尾的最大宽度坡。由于单调栈的下标入栈顺序为下标递增顺序，因此越接近栈底的下标越小，和 $j$ 组成的坡的宽度也越大。为了得到以下标 $j$ 结尾的最大宽度坡，应在栈内找到最小的下标 $i$ 使得 $\text{nums}[i]\le \text{nums}[j]$，因此应该将全部满足 $\text{nums}[i]\le \text{nums}[j]$ 的下标 $i$ 出栈，在出栈的同时更新坡的最大宽度。

2. 假设存在下标 $k$ 满足 $k<j$ 且 $\text{nums}[k]\le \text{nums}[j]$，则任何以 $k$ 结尾的坡的开始下标都可以是以 $j$ 结尾的坡的开始下标，因此以 $k$ 结尾的坡的最大宽度一定小于以 $j$ 结尾的坡的最大宽度。

3. 假设存在下标 $k$ 满足 $k<j$ 且 $\text{nums}[k]>\text{nums}[j]$，则可能存在下标 $p$ 满足 $p<k$ 且 $\text{nums}[j]<\text{nums}[p]\le \text{nums}[k]$，此时下标 $p$ 可以是以 $k$ 结尾的坡的开始下标，但是不可以是以 $j$ 结尾的坡的开始下标。在遍历到 $k$ 时，计算以 $k$ 结尾的最大宽度坡一定会将 $p$ 出栈。

代码

class Solution {
    public int maxWidthRamp(int[] nums) {
        int maxWidth = 0;
        Deque<Integer> stack = new ArrayDeque<Integer>();
        int length = nums.length;
        for (int i = 0; i < length; i++) {
            int num = nums[i];
            if (stack.isEmpty() || nums[stack.peek()] > num) {
                stack.push(i);
            }
        }
        for (int j = length - 1; j > 0; j--) {
            int num = nums[j];
            while (!stack.isEmpty() && nums[stack.peek()] <= num) {
                int width = j - stack.pop();
                maxWidth = Math.max(maxWidth, width);
            }
        }
        return maxWidth;
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是数组 $\text{nums}$ 的长度。需要从左到右遍历数组 $\text{nums}$ 将下标入单调栈，然后从右到左遍历数组 $\text{nums}$ 计算最大宽度。由于每个下标最多入栈和出栈各一次，因此时间复杂度是 $O(n)$。

• 空间复杂度：$O(n)$，其中 $n$ 是数组 $\text{nums}$ 的长度。空间复杂度主要取决于栈空间，栈内元素个数不会超过 $n$。