题意:一排带有颜色的砖块(n<=200),每一个可以消除相同颜色的砖块,每一次可以到块数k的平方分数。问怎么消能使分数最大。
这道题的dp状态的设法非常巧妙。
思路:dp[l][r][k] 表示,从l到r块且r后面有k块与第r快的颜色一样。
dp[l][r][k] = max(dp[l][r][k],dfs(l,i,k+1)+dfs(i+1,r-1,0)); i=1....len
const int maxn=205;
int n;
int dp[maxn][maxn][maxn],a[maxn];
struct node
{
int color,len;
}s[maxn];
int solve(int l,int r,int len)
{
if(dp[l][r][len]!=0) return dp[l][r][len];
int ans,maxn;
ans=(s[r].len+len)*(s[r].len+len);
if(l==r)
{
dp[l][r][len]=ans;
return ans;
}
ans+=solve(l,r-1,0);
for(int i=r-1;i>=l;i--)
{
if(s[i].color!=s[r].color) continue;
maxn=solve(l,i,len+s[r].len)+solve(i+1,r-1,0);
if(maxn<=ans) continue;
ans=maxn;
break;
}
dp[l][r][len]=ans;
return ans;
}
int main()
{
int T;
int Case=0;
scanf("%d",&T);
while(T--)
{
int id=0;
scanf("%d",&n);
for(int i=0;i<n;i++) scanf("%d",&a[i]);
memset(dp,0,sizeof(dp));
s[id].color=a[id];
s[id].len=1;
for(int i=1;i<n;i++)
{
if(a[i]!=a[i-1])
{
s[++id].color=a[i];
s[id].len=1;
}
else ++s[id].len;
}
printf("Case %d: %d\n",++Case,solve(0,id,0));
}
}