单调栈结构

单调栈结构

题目描述

给定一个不含有重复值的数组 arr，找到每一个 i 位置左边和右边离 i 位置最近且值比 arr[i] 小的位置。返回所有位置相应的信息。

输入描述：

第一行输入一个数字 n，表示数组 arr 的长度。

以下一行输出 n个数字，表示数组的值。

输出描述：

输出n行，每行两个数字 L 和 R，如果不存在，则值为-1，下标从0开始。

示例1

输入

7
3 4 1 5 6 2 7

输出

-1 2
0 2
-1 -1
2 5
3 5
2 -1
5 -1

备注：

$1\le n\le 1000000$
$-1000000\le ar{r}_i\le 1000000$

---

题解：

明显的单调栈。从左往右遍历数组，确保栈中的元素从栈顶到栈底满足严格递减结构就行。

举个例子（左边最小值为例），栈中的元素为 {4, 3}，对于大于 3 的元素，4 永远不可能称为最近小于 a[i] 的值，所以需要在栈中维护一个递减关系，以去除无用的元素。

代码：

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 1000001;

int n, a[N];
int stk[N];
int mn_ret[N], mx_ret[N];

int main(void) {
    scanf("%d", &n);
    int top = -1;
    for ( int i = 0; i < n; ++i ) {
        scanf("%d", a + i);
        while ( top >= 0 && a[stk[top]] > a[i] ) --top;
        if ( top == -1 ) mn_ret[i] = -1;
        else mn_ret[i] = stk[top];
        stk[++top] = i;
    }
    top = -1;
    for ( int i = n - 1; i >= 0; --i ) {
        while ( top >= 0 && a[stk[top]] > a[i] ) --top;
        if ( top == -1 ) mx_ret[i] = -1;
        else mx_ret[i] = stk[top];
        stk[++top] = i;
    }
    for ( int i = 0; i < n; ++i )
        printf("%d %d\n", mn_ret[i], mx_ret[i]);
    return 0;
}

简洁写法：

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 1000001;
const int INF = 0x3f3f3f3f;

int n, a[N];
int stk[N];
int left[N], right[N];

int main(void) {
    scanf("%d", &n);
    a[0] = ~INF;
    int top = 1;
    for ( int i = 1; i <= n; ++i ) scanf("%d", a + i);
    for ( int i = 1; i <= n; ) {
        if ( a[i] > a[stk[top - 1]] ) {
            left[i] = stk[top - 1];
            stk[top++] = i++;
        } else right[stk[--top]] = i;
    }
    for ( int i = 1; i <= n; ++i )
        printf("%d %d\n", left[i] - 1, right[i] - 1);
    return 0;
}