poj 1390 blocks

刘汝佳黑书上的题目。书上解释的很详细了，不说了。

#include<iostream>
#include<cstring>
using namespace std;
int dp[205][205][205];
int color[205],len[205];
int maxi(int a,int b)
{
	if(a>b)
		return a;
	return b;
}
int dfs(int start,int end,int last)
{
	if(dp[start][end][last]>=0)
		return dp[start][end][last];
	if(start>end)
	{
		dp[start][end][last]=0;
		return dp[start][end][last];
	}
	if(start==end)
	{
		dp[start][end][last]=(len[end]+last)*(len[end]+last);
		return dp[start][end][last];
	}
	int ans=dfs(start,end-1,0)+(len[end]+last)*(len[end]+last);
	for(int x=end-1;x>=start;x--)
		if(color[x]==color[end])
			ans=maxi(ans,dfs(start,x,len[end]+last)+dfs(x+1,end-1,0));
	dp[start][end][last]=ans;
	return dp[start][end][last];
}
int main()
{
	int test,count,n,length,i,a,j,k;
	cin>>test;
	count=0;
	while(test--)
	{
		count=count+1;
		color[0]=0;
		length=1;
		cin>>n;
		cin>>color[1];
		len[1]=1;
		for(i=2;i<=n;i++)
		{
			cin>>a;
			if(a!=color[length])
			{
				length=length+1;
				color[length]=a;
				len[length]=1;
			}
			else len[length]+=1;
		}
		memset(dp,-1,sizeof(dp));
		int ko=dfs(1,length,0);
		cout<<"Case "<<count<<": "<<ko<<endl;
	}
	return 0;
}