记录编号	247367	评测结果	AAAAAAAAAA
题目名称	[USACO Mar09]餐厅清扫	最终得分	100
用户昵称	TenderRun	是否通过	通过
代码语言	C++	运行时间	0.144 s
提交时间	2016-04-08 20:30:42	内存使用	2.69 MiB

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
const int maxn=100010;
int a[maxn],f[maxn],nxt[maxn],pre[maxn],head[maxn],n,m,k;
int l[maxn],r[maxn];
int main(){
	freopen("cleanup.in","r",stdin);
	freopen("cleanup.out","w",stdout);
	scanf("%d%d",&n,&k);
	memset(f,127,sizeof(f));f[0]=0;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<=n;i++){
		pre[i]=head[a[i]];
		nxt[pre[i]]=i;
		head[a[i]]=i;
		l[i]=i-1;
		r[i]=i+1;
	}	
	for(int i=1;i<=n;i++){
		int ret=0;
		m=ceil(sqrt(i));
		for(int j=i;j>=1;j=l[j]){
			if(ret>m)break;
			if(nxt[j]>i||!nxt[j])ret++;
			else l[r[j]]=l[j],r[l[j]]=r[j];			
			f[i]=min(f[i],f[l[j]]+ret*ret);	
		}
	}
	//for(int i=1;i<=n;i++)
		printf("%d\n",f[n]);
	return 0;	
}