【poj1390】Blocks【区间dp】

传送门

一道很nice的题。首先肯定将颜色相同的缩点，因为中间不可能插进去不同的。

首先可以很明白的看出是区间动态规划。但是接下来我们发现不太清楚怎么转移，因为中间和两边选的顺序可能不一样。

我们考虑最右边这个大联通块有多少个颜色一样的。因为有些可能是别人并过来的。

然后策略就分两种。一种是直接把最右边的吃了。另一种是把中间的吃了然后把右边的弄过来整成一块。

设f(l,r,k)表示这个区间右边还有k个和r颜色相同的。

首先f(l,r,k)=f(l,r-1,0)+k*k;

所以我们枚举中间和r颜色相同的位置x，得到

f(l,r,k)=max(f(l,r,k),f(l,x,k+len[r])+f(x+1,r-1,0);

如此就可以直接记忆化了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define in read()
int in{
	int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){
		ch=getchar();if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar();
	}return cnt*f;
}
int f[203][203][203];
int t,n,a[203];
int cnt,color[203],len[203];
int cse;
int sqr(int x){
	return x*x;
}
int dfs(int l,int r,int k){
	if(f[l][r][k]!=-1)return f[l][r][k];
	if(l==r)return sqr(len[r]+k);
	int gu=dfs(l,r-1,0)+sqr(len[r]+k);
	for(int i=l;i<r;i++){
		if(color[i]==color[r]){
			gu=max(gu,dfs(i+1,r-1,0)+dfs(l,i,len[r]+k));
		}
	}return f[l][r][k]=gu;
}
signed main(){
	t=in;
	while(t--){
		n=in;cnt=0;memset(f,-1,sizeof(f));
		for(int i=1;i<=n;i++)a[i]=in;
		for(int i=1;i<=n;i++){
			if(a[i]!=a[i-1]){
				color[++cnt]=a[i];len[cnt]=1;
			}else len[cnt]++;
		}
		++cse;
		cout<<"Case "<<cse<<": "<<dfs(1,cnt,0)<<'\n';
	}
	return 0;
}