Codeforces Round #622 (Div. 2) C2

C2. Skyscrapers (hard version)

相比于C1我们暴力找峰值 我们优化转移
考虑递推
l[i]表示以i作为峰值左边所有呈单调增图形的值
r[i]同
考虑这个的转移 要是左边一个比他高肯定不从比他高的转移
我们画个图会发现转移是找到左边第一个比他低的点来转移 这个可以用单调栈维护
转移就是 l[i]=l[j]+（i-j)*a[i]
然后最后ans=max{l[i]+r[i]-a[i]}

void solve(){
    int n;cin>>n;
    vector<int>a(N),R(N),L(N),l(N),r(N);
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    stack<int>stk;
    a[n + 1] = -INF;
    for (int i = 1; i <= n + 1; i++) {
        while (stk.size() && a[stk.top()] > a[i]) {
            R[stk.top()] = i;
            stk.pop();
        }
        stk.push(i);
    }
    while (stk.size())stk.pop();
    a[0] = -INF;
    for (int i = n; i >= 0; i--) {
        while (stk.size() && a[stk.top()] > a[i]) {
            L[stk.top()] = i;
            stk.pop();
        }
        stk.push(i);
    }
    for(int i=1;i<=n;i++){
        l[i]=l[L[i]]+(i-L[i])*a[i];
    }
    for(int i=n;i>=1;i--){
        r[i]=r[R[i]]+(R[i]-i)*a[i];
    }
    int ans=-INF,pos;
    for(int i=1;i<=n;i++){
        if(l[i]+r[i]-a[i]>ans){
            ans=l[i]+r[i]-a[i];
            pos=i;
        }
    }
    vector<int>res=a;
    for(int i=pos-1;i>=1;i--){
        res[i]=min(res[i+1],res[i]);
    }
    for(int i=pos+1;i<=n;i++){
        res[i]=min(res[i-1],res[i]);
    }
    for(int i=1;i<=n;i++)cout<<res[i]<<' ';
}