题意:
给一行有颜色的方块,相同颜色且相邻的方块可以消去并得到(该颜色长度)^2的分数,消去一些方块后剩余方块会自动靠拢,求消去所有方块的最高分数。
分析:
黑书上的例题,类似区间dp一般是dp[i][j]表示区间状态(i,j)能取到的最优值,这题用dp[i][j][k],这个k看似与最优值没啥关系。。仔细一想却很巧妙,他假设之后会有k个与第j段颜色相同的方块一起消,这样在后面某段的计算需要和前面一起消时就能直接调用结果了。
代码:
//poj 1390
//sep9
#include <iostream>
using namespace std;
const int maxN=205;
int dp[maxN][maxN][maxN];
int color[maxN],len[maxN];
int n;
void search(int i,int j,int k)
{
if(dp[i][j][k]) return ;
if(i==j){
dp[i][j][k]=(len[i]+k)*(len[i]+k);
return ;
}
if(i>j) return ;
search(i,j-1,0);
dp[i][j][k]=dp[i][j-1][0]+(len[j]+k)*(len[j]+k);
for(int x=i;x<j;++x){
if(color[x]!=color[j]) continue;
search(i,x,k+len[j]);
search(x+1,j-1,0);
dp[i][j][k]=max(dp[i][j][k],dp[i][x][k+len[j]]+dp[x+1][j-1][0]);
}
return ;
}
int main()
{
int i,x,p,cases,t=0;
scanf("%d",&cases);
while(cases--){
scanf("%d",&n);
color[0]=INT_MIN;
for(p=0,i=1;i<=n;++i){
scanf("%d",&x);
if(x!=color[p]){
++p;
color[p]=x;
len[p]=1;
}
else
++len[p];
}
memset(dp,0,sizeof(dp));
search(1,p,0);
printf("Case %d: %d\n",++t,dp[1][p][0]);
}
return 0;
}