bzoj4709 [jsoi2011]柠檬

Description

Flute 很喜欢柠檬。它准备了一串用树枝串起来的贝壳，打算用一种魔法把贝壳变成柠檬。贝壳一共有 N (1 ≤ N

 ≤ 100,000) 只，按顺序串在树枝上。为了方便，我们从左到右给贝壳编号 1..N。每只贝壳的大小不一定相同，

贝壳 i 的大小为 si(1 ≤ si ≤10,000)。变柠檬的魔法要求，Flute 每次从树枝一端取下一小段连续的贝壳，并

选择一种贝壳的大小 s0。如果 这一小段贝壳中 大小为 s0 的贝壳有 t 只，那么魔法可以把这一小段贝壳变成 s

0t2 只柠檬。Flute 可以取任意多次贝壳，直到树枝上的贝壳被全部取完。各个小段中，Flute 选择的贝壳大小 s

0 可以不同。而最终 Flute 得到的柠檬数，就是所有小段柠檬数的总和。Flute 想知道，它最多能用这一串贝壳

变出多少柠檬。请你帮忙解决这个问题。

Input

第 1 行：一个整数，表示 N。

第 2 .. N + 1 行：每行一个整数，第 i + 1 行表示 si。

Output

仅一个整数，表示 Flute 最多能得到的柠檬数。

对位置i，f[i]表示考虑前i个元素的最多柠檬数

f[0]=1

f[i]=max f[j-1]+a[i]*(j到i之间si出现次数)^2 , 1<=j<=i,si=sj

对每个si分别用单调栈维护斜率优化dp，决策不单调但可以三分求出决策点

#include<cstdio>
#include<vector>
typedef long long i64;
const int R=,N=;
char buf[R+],*ptr=buf-;
int _(){
    int x=,c=*++ptr;
    while(c<)c=*++ptr;
    while(c>)x=x*+c-,c=*++ptr;
    return x;
}
int n,a[N],pv[N],pw[],c[N];
i64 f[N];
std::vector<int>q[];
void maxs(i64&a,i64 b){if(a<b)a=b;}
i64 p2(i64 x){return x*x;}
double F(int x,int y,int ci,int ai){
    return (f[x-]-f[y-]+ai*(p2(c[x])-p2(c[y])))/double(i64(c[x]-c[y])*ai<<);
}
int main(){
    fread(buf,,R,stdin);
    n=_();
    for(int i=;i<=n;++i){
        pv[i]=pw[a[i]=_()];
        c[i]=c[pw[a[i]]]+;
        pw[a[i]]=i;
    }
    for(int i=;i<=n;++i){
        int w=a[i],ci=c[i]+;
        std::vector<int>&Q=q[w];
        while(Q.size()>&&F(i,Q[Q.size()-],ci,w)>F(Q[Q.size()-],Q[Q.size()-],ci,w))Q.pop_back();
        Q.push_back(i);
        int l=,r=Q.size()-;
        while(l+<=r){
            int m1=l+r>>,m2=m1+r>>;
            if(f[Q[m1]-]+w*p2(ci-c[Q[m1]])<f[Q[m2]-]+w*p2(ci-c[Q[m2]]))l=m1;
            else r=m2;
        }
        for(int p=l;p<=r;++p)maxs(f[i],f[Q[p]-]+w*p2(ci-c[Q[p]]));
    }
    printf("%lld",f[n]);
    return ;
}