题意:给n(小于200)个砖块,每个砖块有自己的颜色,你每次可以选择一段颜色相等的砖块然后把它消去,得分为长度的平方,问你这段砖块最多能得多少分。
分析:黑书上的经典DP,以前写过,今天趁着忘了又试着写了写,然后发现自己怎么个搞不出来状态的表示方法。。。
f[i,j,k]表示 [i,j]这一段砖块后边再接上长度为k的颜色和第j块一样的砖块的最高得分,然后我们发现只要讨论第j块砖最后和谁合并就可以转移了。
#include<iostream>
#include<string>
#include<algorithm>
#include<cstdlib>
#include<cstdio>
#include<set>
#include<map>
#include<vector>
#include<ctime>
#include<cstring>
#include<stack>
#include<cmath>
#include<queue>
#define INF 0x3f3f3f3f
#define eps 1e-9
#define N 205
using namespace std;
int n,T,a[N],dp[N][N][N];
int dfs(int l,int r,int k)
{
if(l > r) return 0;
if(l == r) return (k+1)*(k+1);
if(dp[l][r][k] > 0) return dp[l][r][k];
dp[l][r][k] = dfs(l,r-1,0)+ (k+1)*(k+1);
for(int i = l;i < r;i++)
if(a[i] == a[r]) dp[l][r][k] = max(dp[l][r][k],dfs(l,i,k+1) + dfs(i+1,r-1,0));
return dp[l][r][k];
}
int main()
{
cin.sync_with_stdio(false);
cin>>T;
for(int t = 1;t <= T;t++)
{
memset(dp,-1,sizeof(dp));
cin>>n;
for(int i = 1;i <= n;i++) cin>>a[i];
cout<<"Case "<<t<<": "<<dfs(1,n,0)<<endl;
}
}