染色游戏

题目描述

蒜头是一名优秀的画家。蒜头有一张长度为$n$的画卷，在位置$i$上画图案会获得$a_i$的美观度。蒜头是一个有追求的人，因此他希望他的画从左往右是越来越美观的，即对于任意两个画了图案的格子$l < r$，有$a_ l ≤ a_ r$。但蒜头发现，人们评判画卷的好坏，并不会只从画出的图案来考虑。具体来说，一张画卷的美观度，定义为所有画了图案的位置的美观度之和与在图上选择两个可以重复的位置使得两位置之间不存在画了图案的位置的方案数之差。现在，蒜头想要知道，他画出的画卷的最大美观度是多少。 形式化地，一段连续的长度为 m 的空白位置会让美观度降低 $\frac {m ( m + 1 )} 2$ 。

输入

输入的第一行是一个数$n$ ，表示序列的长度。
接下来一行 $n$ 个数，第 i 个数表示$a_i(-10^8<ai<10^8)$ 。

输出

输出一行表示最大美观度。

题解

如果不考虑$a_ l ≤ a_ r$的条件，因为$a_i$可以是负数所以有dp式

$$dp[i]=max_{j<i}(dp[j]+a[i]-\frac{(i-j-1)(i-j)}2)$$ $dp[i]$表示只选前 $i$个并且一定选第$i$时最大的美观度，这是 $O(n^2)$的效率。打开这个式子，整理后不难发现 $$dp[i]=dp[j]+a[i]-i^2/2+ij/2+ij/2-j^2/2+i/2-j/2$$ $$-dp[j]+j^2/2+j/2=ij-dp[i]+a[i]-i^2/2+i/2$$ 于是得到了斜率优化的式子，dp[i]最大即截距最小，于是我们维护下凸壳即可。
但是我们现在回过头来思考 $a_ l ≤ a_ r$的条件，因为 $j<i$所以 $a_j\leq a_i$即 $i$的状态是由$j$转移过来的，于是我们想按值来排序，但是下标就被破坏了，于是CDQ分治的思想闪亮登场。
我们中序分治处理(保证单调性)，先按值排序一遍，这样每次左区间的值一定是小于右区间的，层层分治时对左右区间分别按下标排序，然后左边取出下标小于右边当前下标的点入单调栈，保证正确性，然后用左边的点更新右边的dp值，排序用快排，总时间效率每层nlogn+n,logn层反正是 $O(nlog^2n)$，更优的方法是先用归并排序预处理出每一层(一般CDQ分治排序和分治一起做了)，这样效率是两个排序一个分治3个 $O(nlogn)$

代码

#include <bits/stdc++.h>
#define maxn 1000005
#define INF 0x3f3f3f3f
typedef long long LL;
using namespace std;
LL n,m;
LL temp[25][maxn],a[maxn],w[maxn],f[maxn],g[maxn],d[maxn],ans=-1e15;
void orz(LL l,LL r,LL t){
    if(l>r) return;
    if(l==r){
        f[w[l]]=max(f[w[l]],a[w[l]]-(w[l]-1)*w[l]/2);
        g[w[l]]=-2*f[w[l]]+w[l]*w[l]+w[l];
        ans=max(ans,f[w[l]]-(n-w[l])*(n-w[l]+1)/2);
        return;
    }
    LL mid=l+r>>1;
    orz(l,mid,t+1);
    for(LL x=l,y=mid+1,p=0,q=0;y<=r;y++){
        LL nowr=temp[t][y];
        while(temp[t][x]<nowr && x<=mid){
            LL now=temp[t][x];
            while(p<q&&(g[d[q]]-g[d[q-1]])*(now-d[q])>=(g[now]-g[d[q]])*(d[q]-d[q-1])) q--;
            d[++q]=now;
            x++;
        }
        while(p<q&&(g[d[p+1]]-g[d[p]]<=2*nowr*(d[p+1]-d[p]))) p++;
        f[nowr]=max(f[nowr],f[d[p]]+a[nowr]-(nowr-d[p])*(nowr-d[p]-1)/2);
        g[nowr]=-2*f[nowr]+nowr*nowr+nowr;
        ans=max(ans,f[nowr]-(n-nowr)*(n-nowr+1)/2);
    }
    orz(mid+1,r,t+1);
}
void orzz(LL l,LL r,LL t){
    if(l>r) return;
    if(l==r) {temp[t][l]=w[l]; return;}
    LL mid=l+r>>1;
    orzz(l,mid,t+1); orzz(mid+1,r,t+1);
    LL p=l,q=mid+1,tail=l;
    while(p<=mid && q<=r){
        if(temp[t+1][p]<temp[t+1][q]) temp[t][tail++]=temp[t+1][p++];
        else temp[t][tail++]=temp[t+1][q++];
    }
    while(p<=mid) temp[t][tail++]=temp[t+1][p++];
    while(q<=r) temp[t][tail++]=temp[t+1][q++];
}
bool cmp(LL x,LL y) {
    return (a[x]<a[y])||(a[x]==a[y]&&x<y);
}
int main(){
    scanf("%lld",&n);
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        w[i]=i;
        f[i]=-1e15;
    }
    ans=-n*(n+1)/2;
    sort(w+1,w+n+1,cmp);
    orzz(1,n,0);
    f[0]=g[0]=0;
    orz(1,n,1);
    printf("%lld\n",ans);
    return 0;
}