想法-Codeforces-1313C2-Skyscrapers (hard version)

题目链接
题意：
n座楼，每个楼有一个最高高度a[i]，要求每座楼左右两侧不能同时有比他高的楼，输出所有楼的高度之和最大时，每座楼的高度
思路：
依次判断每座楼为最高的楼时，所有楼的最大高度之和，记录最大值最高楼的位置，然后依次得出每座楼的高度
对于每座楼，需要找到它左边最大高度之和及右边最大高度之和
a[i]为最高楼时，如果a[i]>=a[i-1]，那么l[i]=l[i-1]+a[i]
如果a[i]<a[i-1]时，那么需要从最左边开始，找到第一个比a[i]高的楼a[x]，把这之间的楼都变成和a[i]一样高，即l[i]=l[x]+(i-x)*a[i],
右侧用同样的方法。
★如何快速找到第一个比a[i]高的楼：
lpos[i]表示从1开始第一座大于a[i]的楼的位置，从i-1开始找，不断往前递推，找到的最后一个满足a[x]>a[i]的x即为从左到右第一个比a[i]大的楼，lpos[i]=x;
代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll l[500005], r[500005];
ll a[500005];
ll lpos[500005], rpos[500005];
ll ans[500005];
int main()
{
    ll n;
    scanf("%lld", &n);
    ll i, j, k;
    for (i = 1; i <= n; i++)
    {
        scanf("%lld", &a[i]);
        if (a[i] >= a[i - 1])
        {
            l[i] = l[i - 1] + a[i];
            lpos[i] = i - 1;
        }
        else
        {
            ll pos = i - 1;
            while (pos >= 1 && a[pos] > a[i])
            {
                pos = lpos[pos];
            }
            lpos[i] = pos;
            l[i] = l[pos] + (i - pos) * a[i];
        }
    }
    a[i + 1] = 0;
    for (i = n; i >= 1; i--)
    {
        if (a[i] >= a[i + 1])
        {
            r[i] = r[i + 1] + a[i];
            rpos[i] = i + 1;
        }
        else
        {
            ll pos = i + 1;
            while (pos >= 1 && a[pos] > a[i])
            {
                pos = rpos[pos];
            }
            rpos[i] = pos;
            r[i] = r[pos] + (pos - i) * a[i];
        }
    }
    ll maxx = -1, pos;
    for (i = 1; i <= n; i++)
    {
        if (l[i] + r[i] - a[i] > maxx)
        {
            maxx = l[i] + r[i] - a[i];
            pos = i;
        }
    }
    ans[pos] = a[pos];
    for (i = pos - 1; i >= 1; i--)
    {
        if (a[i] > ans[i + 1])
            ans[i] = ans[i + 1];
        else
            ans[i] = a[i];
    }
    for (i = pos + 1; i <= n; i++)
    {
        if (a[i] > ans[i - 1])
            ans[i] = ans[i - 1];
        else
            ans[i] = a[i];
    }
    for (i = 1; i <= n; i++)
        printf("%lld ", ans[i]);
    printf("\n");
}