CF1313C2 Skyscrapers (hard version) （单调栈）

Skyscrapers (hard version)

思路：我们需要维护当下标为inx的大楼为最高的时候，两边大楼的情况。我们可以把下标为inx的大楼最高时，分别统计左边和右边的情况，我们可以用单调栈维护最小值（小->大），如果栈顶的元素的高度小于inx的大楼，说明之前的所有大楼都因为栈顶的大楼而小于了inx的大楼的高度，如果栈顶的大楼高度大于等于inx的大楼，则说明栈顶下面第一个的大楼到栈顶的大楼之间的大楼需要改变为inx大楼的高度，需要注意空栈的情况。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <functional>
#include <set>
#include <vector>
#include <queue>
#include <cstring>
#include <stack>
 
using namespace std;
 
#define ll long long
#define pb push_back
#define fi first
#define se second
 
 
void solve(){
    int n;
    cin >> n;    
    vector<int > a(n);
    for(auto& in : a) cin >> in;
    stack<int > Sta;
    vector<ll > pre(n);
    for(int now = 0; now < n; ++now){
        while(!Sta.empty()){
            int top = Sta.top();
            if(a[top] >= a[now]){
                Sta.pop();
                if(!Sta.empty()){
                    pre[now] += (ll)a[now] * (top - (int)Sta.top());
                }else {
                    pre[now] += (ll)a[now] * (top + 1);
                }
            }else {
                pre[now] += pre[top];
                break;
            }
        }
        Sta.push(now);
        pre[now] += a[now];
    }
    // for(int i = 0; i < n; ++i) cout << pre[i] << " ";
    // cout << endl;
    vector<ll > rpre(n);
    while(!Sta.empty()) Sta.pop();
    vector<int > cp(a);
    reverse(a.begin(), a.end());
    for(int now = 0; now < n; ++now){
        while(!Sta.empty()){
            int top = Sta.top();
            if(a[top] >= a[now]){
                Sta.pop();
                if(!Sta.empty()){
                    rpre[now] += (ll)a[now] * (top - (int)Sta.top());
                }else {
                    rpre[now] += (ll)a[now] * (top + 1);
                }
            }else {
                rpre[now] += rpre[top];
                break;
            }
        }
        Sta.push(now);
        rpre[now] += a[now];
    }
    // for(auto& x : rpre) cout << x << " ";
    // cout << endl;
    int inx = 0;
    //cp[i]被算了两次，需要减去一次
    for(int i = 1; i < n; ++i){
        inx = (pre[i] + rpre[n - 1 - i] - cp[i] > pre[inx] + rpre[n - 1 - inx]  - cp[inx] ? i : inx);
    }
    // cout << "inx = " << inx << endl;
    int Min = cp[inx];
    vector<int > ans(n);
    for(int i = inx; i >= 0; --i){
        Min = min(Min, cp[i]);
        ans[i] = Min;
    }
    Min = cp[inx];
    for(int i = inx + 1; i < n; ++i){
        Min = min(Min, cp[i]);
        ans[i] = Min;
    }
    for(auto x : ans) cout << x << " ";
    cout << endl;
 
}
 
int main(){
    
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    solve();
    
    return 0;
}