bzoj:1584: [Usaco2009 Mar]Cleaning Up 打扫卫生

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

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]);
}