AcWing 830. 单调栈

题目描述

---

分析：

本题是单调栈经典且少数的应用。维护栈内元素的单调性是有助于求解左边/右边第一个比它小/大的数这样的问题，与暴力双层 for 循环相比能够大量减少比较次数。我们先看一下暴力解的方法：

for (int i = 0; i < n; i ++)
	for (int j = i - 1; j >= 0; j --)
		// compare and select

很明显这是一个 $O(n^2)$ 的复杂度。在内层的 for 循环中，有时会浪费大量时间。比如看下面的分析：

假设输入为 $S_k$，对其分析有两种情况：

1. 若 $S_k＞S_{k-1}$，输出 $S_{k-1}$ 即可。
2. 若 $S_k\le S_{k-1}$，要考虑的就多了！
   ① 当 $S_k\le S_{k-1}$ 且 $S_k\le [S_{k-1}\cdot \cdot \cdot \cdot \cdot \cdot S_1]$ 时($S_{k-1}\cdot \cdot \cdot \cdot \cdot \cdot S_1$ 大小关系并非具有单调性，可以对其任意一个元素都满足这种小于关系)。对于输入的 $S_k$，要输出 $-1$。
   ② 当 $S_k\le S_{k-1}$ 但 $S_k$ 大于 [$S_{k-2}\cdot \cdot \cdot \cdot \cdot \cdot S_1$] 中的某一个元素时。记为 $S_i$，说明 $S_k\le [S_{k-1}\cdot \cdot \cdot \cdot \cdot \cdot S_{i+1}]$ 的全部元素，但大于 $S_i$，此时应输出 $S_i$ 。

请务必搞明白输入 $S_k$ 时 2.>② 中的内容，这是理解下面内容的关键。

假设输入为 $S_{k+1}$，对其分析有两种情况：

1. 若 $S_{k+1}＞S_k$，输出 $S_k$ 即可。
2. 若 $S_{k+1}\le S_k$，要考虑的就更多了！
   ① 当 $S_{k+1}\le S_k$ 且 $S_{k+1}\le [S_k\cdot \cdot \cdot \cdot \cdot \cdot S_1]$，对于输入的 $S_{k+1}$，要输出 $-1$。
   ② 当 $S_{k+1}\le S_k$ 但 $S_{k+1}$ 大于 [$S_{k-1}\cdot \cdot \cdot \cdot \cdot \cdot S_1$] 中的某一个元素时。记为 $S_j$，说明 $S_k\le [S_k\cdot \cdot \cdot \cdot \cdot \cdot S_{j+1}]$ 的全部元素，但大于 $S_j$，此时应输出 $S_j$ 。

我们来仔细分析一下输入 $S_{k+1}$ 时 2.>② 中的内容：

那么怎么对其优化呢？由上图我们可以得知 $S_{k+1}\le [S_k\cdot \cdot \cdot \cdot \cdot \cdot S_{i+1}]$，即 $[S_k\cdot \cdot \cdot \cdot \cdot \cdot S_{i+1}]$ 皆不能作为输入值为 $S_{k+1}$ 的输出，而在暴力做法中在这里会消耗大量的比较时间。这就是单调栈改进的地方，在不断出栈的过程中维护一个“合理”的区间。这要求在输入为 $S_k$ 时进行维护，而在输入为 $S_{k+1}$ 时享受这一成果。
具体步骤为：输入为 $S_k$ 时，若栈不为空，将栈中比 $S_k$ 大的元素全部出栈。这一定是有效的，因为：
若 $S_{k+1}$ 大于 $S_k$，输出 $S_k$。
若 $S_{k+1}$ 小于等于 $S_k$，那将更会小于出栈的元素，比了也没用！

---

代码(C++)

#include <iostream>

using namespace std;

const int N = 100010;
int stk[N], top; // top 为 0 时，栈为空

int main()
{
    int n;
    cin >> n;
    
    for (int i = 0; i < n; i ++)
    {
        int k;
        cin >> k;
        
        // 当栈非空且栈顶元素大于等于输入元素时，不断出栈，为下一个输入元素“谋福利”
        while (top && stk[top] >= k) top --;
        
        // 若栈非空，则栈顶元素为左边第一个小于输入 x 的元素
        if (top) cout << stk[top] << ' ';
        // 栈为空时没有目标值
        else cout << "-1 ";
        
        // 将当前值入栈作为后面输入的比较元素
        stk[++ top] = k;
    }
    return 0;
}

---