Description
有N头奶牛，每头那牛都有一个标号Pi，1 <= Pi <= M <= N <= 40000。现在Farmer John要把这些奶牛分成若干段，定义每段的不河蟹度为：若这段里有k个不同的数，那不河蟹度为k*k。那总的不河蟹度就是所有段的不河蟹度的总和。
Input
第一行：两个整数N，M
第2..N+1行：N个整数代表每个奶牛的编号
Output
一个整数，代表最小不河蟹度
Sample Input
13 4
1
2
1
3
2
2
3
4
3
4
3
1
4
1
2
1
3
2
2
3
4
3
4
3
1
4
Sample Output
11
在n^2的转移中，只有k*k<=n的那些k有用，所以就n^(3/2)直接丢了。
/************************************************************** Problem: 1584 User: JSZX11556 Language: C++ Result: Accepted Time:128 ms Memory:1604 kb ****************************************************************/ #include<queue> #include<cstdio> #include<algorithm> #define MN 40001 using namespace std; int read_p,read_ca,read_f; inline int read(){ read_p=0;read_ca=getchar();read_f=1; while(read_ca<'0'||read_ca>'9') read_f=read_ca=='-'?-1:read_f,read_ca=getchar(); while(read_ca>='0'&&read_ca<='9') read_p=read_p*10+read_ca-48,read_ca=getchar(); return read_p*read_f; } int n,m,a[MN],t[MN],mmh[MN],ne[MN],fi[MN]; int main(){ register int i,j; n=read();m=read(); for (i=1;i<=n;i++) a[i]=read(); for (i=1;i<=n;i++){ if (t[a[i]]) fi[ne[a[i]]]=fi[a[i]],ne[fi[a[i]]]=ne[a[i]]; t[a[i]]=i;ne[a[i]]=ne[0];fi[ne[0]]=a[i];ne[0]=a[i];fi[a[i]]=0; mmh[i]=1e9; for (j=1,m=ne[a[i]];j*j<=n&&m;j++,m=ne[m]) mmh[i]=min(mmh[i],mmh[t[m]]+j*j); if (!m) mmh[i]=min(mmh[i],mmh[0]+j*j); } printf("%d ",mmh[n]); }