CodeForces - 1313C2 Skyscrapers（单调栈）

题目链接：https://vjudge.net/problem/CodeForces-1313C2

题意：给你n个数，每个数代表每个位置所能到达的最大值。现在让你构造出左边单调递增，右边单调递减且和最大的序列并输出。多种答案，输出其中任意一个就好。

思路：首先考虑最暴力的解法，就是两层for循环直接暴力枚举每一个位置作为最大值的情况。但这一题5e5的数据范围肯定会超时，所以我们考虑一下其他的方法。我们考虑一下这一题暴力方法第二层for循环的作用就是求解第i个位置作为最高点时可以取得的最大值，如果我们可以提前预处理出来，那这一题我们就可以用一层for循环，即O(n)得时间复杂度求解出来。那该如何预处理呢？我们考虑用两个数组dp_l[]存储以i作为最高点往左边可以取得得最大的和，dp_r[]存储以i作为最高点往右可以取得的最大的和。这样第i个位置作为最高点可以取得的最大的和就是dp_l[i] + dp_r[i] - a[i]。那这两个数组如何求解呢？以dp_l为例（dp_r同理），我们先找出位置pos表示i左边第一个比a[i]小的数的位置，那
dp_l[i] = dp_l[pos] + (i - pos) * a[i]。因为我们要以i为最大值，所以我们把i左边比a[i]大的全部变成a[i]肯定是最优的，之后我们再加上前面已经处理出来的dp_l[pos]就好啦。所以我们现在只要找出这个位置pos就好啦，这个就是单调栈的模板题啦。具体的详见代码。

AC代码：

#include <stack>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define LL long long

const int maxn = 5e5 + 7;

struct node {
    LL val;
    int l,r;
} a[maxn];

stack<int>s;

///单调栈预处理
void init(int n) {
    while(!s.empty()) s.pop();
    for(int i = 1 ; i <= n ; i++) {
        while(!s.empty() && a[s.top()].val > a[i].val) s.pop();
        if(s.empty()) a[i].l = 1;
        else a[i].l = s.top() + 1;
        s.push(i);
    }
    while(!s.empty()) s.pop();
    for(int i = n ; i >= 1 ; i--) {
        while(!s.empty() && a[s.top()].val > a[i].val) s.pop();
        if(s.empty()) a[i].r = n;
        else a[i].r = s.top() - 1;
        s.push(i);
    }
}

LL dp_l[maxn],dp_r[maxn];
LL ans[maxn]; ///记录答案

int main() {
    int n;
    while(~scanf("%d",&n)) {
        for(int i = 1 ; i <= n ; i++) {
            scanf("%lld",&a[i].val);
            ans[i] = a[i].val;
        }
        init(n);
        memset(dp_l,0,sizeof(dp_l));
        memset(dp_r,0,sizeof(dp_r));
        for(int i = 1 ; i <= n ; i++) {
            dp_l[i] = dp_l[a[i].l-1] + (i - a[i].l + 1) * a[i].val;
        }
        for(int i = n ; i >= 1 ; i--) {
            dp_r[i] = dp_r[a[i].r+1] + (a[i].r - i + 1) * a[i].val;
        }
        int temp = 0;
        LL ANS = 0;
        for(int i = 1 ; i <= n ; i++) {
            LL res = dp_l[i] + dp_r[i] - a[i].val;
            if(res > ANS) {
                ANS = res;
                temp = i;
            }
        }
        for(int i = temp + 1 ; i <= n ; i++) {
            ans[i] = min(ans[i-1],ans[i]);
        }
        for(int i = temp - 1 ; i >= 1 ; i--) {
            ans[i] = min(ans[i+1],ans[i]);
        }
        for(int i = 1 ; i <= n ; i++) {
            if(i != 1) printf(" ");
            printf("%d",ans[i]);
        }
        printf("\n");
    }
    return 0;
}