[JSOI 2011]柠檬

Description

题库链接

给你一个长度为 (n) 的序列 (s_i)，你可以将他分为若干段，每一段的价值为 (st^2)，其中 (s) 表示你指定的一个数，(t) 表示这个数在这一段中出现的次数。你需要最大化价值和。

(1leq nleq 100000,1leq s_ileq 10000)

Solution

首先最优分出来的这一段两端一定是颜色相同的。

于是我们可以按颜色来 DP，设 (f_i=maxlimits_{jleq i,s_j=s_i}{f_{j-1}+s_i(sum_i-sum_j+1)^2})。其中 (sum) 表示该颜色个数的前缀和。

这个平方式子是可以斜率优化的，把平方拆开后可以化成 (y=kx+b)。因为需要最大化 (f_i)，那么就要最小化截距。由于 (k) 是单调递减的，因此要维护一个下凸包，并且决策点在凸包上往左移。

（其实就是四边形不等式 (w) 是满足 (geq) 的，并且 DP 值取 (max)。原理戳这）

因此用单调栈维护即可。

Code

#include <bits/stdc++.h>
#define ll long long
#define x(a) (s[a])
#define y(a) (-f[a-1]-1ll*t*s[a]*s[a]+2ll*t*s[a])
#define pop pop_back
#define push push_back
using namespace std;
const int N = 100000+5;

int n, l[N], c[N], s[N];
vector<int> S[N];
ll f[N];

int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++)
        scanf("%d", &c[i]), s[i] = s[l[c[i]]]+1, l[c[i]] = i;
    for (int i = 1; i <= n; i++) {
        int t = c[i], sz;
        while ((sz = S[t].size()) >= 2 &&
         (y(S[t][sz-1])-y(i))*(x(S[t][sz-2])-x(i)) <= (y(S[t][sz-2])-y(i))*(x(S[t][sz-1])-x(i))) S[t].pop();
        S[t].push(i);
        while ((sz = S[t].size()) >= 2 && (y(S[t][sz-1])-y(S[t][sz-2])) >= -2ll*t*s[i]*(x(S[t][sz-1])-x(S[t][sz-2]))) S[t].pop();
        f[i] = f[S[t][(sz = S[t].size())-1]-1]+1ll*t*(s[i]-s[S[t][sz-1]]+1)*(s[i]-s[S[t][sz-1]]+1);
    }
    printf("%lld
", f[n]);
    return 0;
}