bzoj4709 [Jsoi2011]柠檬（贪心+dp+决策单调性+单调栈）

问题就是把这个序列分成若干段，使得每段的贡献和最大。
首先我们贪心地发现每段的起终位置一定相同（否则分出去更好）
于是有dp，f[i]表示前i个的最大贡献。枚举i这一段的起点j，一定满足a[i]==a[j],记c[i]表示i及i之前出现了多少个a[i]
$f[i]=\max\limits_{j=1}^{i}\{f[j-1]+a[i]*(c[i]-c[j]+1)^2|a[j]==a[i]\}$
复杂度$O(n^2)$
考虑进一步优化。
设k1< k2且k1优于k2
那么推一推可以发现如果k1在某时刻优于k2，那么以后都有k1优于k2。
于是我们可以单调栈维护一波决策点，保证从栈顶往下的每个点优于上一个点的时间单增。
k1优于k2的最早时间i可以二分来求。
于是复杂度$O(nlogn)$

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,cnt[10010],a[N],c[N];
ll f[N];
vector<int>qq[10010];
inline ll sqr(ll x){return x*x;}
inline ll calc(int j,int ci){return f[j-1]+(ll)a[j]*sqr(ci-c[j]+1);}
inline int beyond(int k1,int k2){//k1优于k2的最早时间i
    int l=c[k2],r=n;
    while(l<=r){
        int mid=l+r>>1;
        if(calc(k1,mid)>=calc(k2,mid)) r=mid-1;
        else l=mid+1;
    }return r+1;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();
    for(int i=1;i<=n;++i){
        int x=a[i]=read();c[i]=++cnt[x];
        while(qq[x].size()>=2&&beyond(*(qq[x].end()-2),qq[x].back())<=beyond(qq[x].back(),i)) qq[x].pop_back();
        qq[x].push_back(i);
        while(qq[x].size()>=2&&beyond(*(qq[x].end()-2),qq[x].back())<=c[i]) qq[x].pop_back();
        f[i]=calc(*(qq[x].end()-1),c[i]);
    }printf("%lld\n",f[n]);
    return 0;
}