题意:给定n个不同颜色的盒子,连续的相同颜色的k个盒子可以拿走,权值为k*k,求把所有盒子拿完的最大权值。
题解:记忆化搜索,dp[ll][rr][kk]代表在区间[ll,rr]取盒子,其中ll前面有k个盒子颜色和rr一样,可以重叠在rr上消去。
那么,dp[ll][rr][kk]=max(dp[ll][i][kk+len[rr]]+dp[i+1,rr-1][0],dp[ll][rr-1][0]+pow(len[rr]+kk)),i是所有位于[ll,rr-1]区间且与rr颜色相同的盒子;
View Code
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 int dp[205][205][205]; 6 int col[205],len[205]; 7 inline int pow(int x) 8 { 9 return x*x; 10 } 11 int dfs(int ll,int rr,int kk) 12 { 13 if(ll>rr) 14 return 0; 15 else if(dp[ll][rr][kk]) 16 return dp[ll][rr][kk]; 17 else if(ll==rr) 18 return dp[ll][rr][kk]=pow(kk+len[rr]); 19 else 20 { 21 int ans=dfs(ll,rr-1,0)+pow(len[rr]+kk); 22 for(int i=rr-1;i>=ll;i--) 23 if(col[i]==col[rr]) 24 ans=max(ans,dfs(ll,i,kk+len[rr])+dfs(i+1,rr-1,0)); 25 return dp[ll][rr][kk]=ans; 26 } 27 } 28 int main() 29 { 30 int T,ca=0; 31 for(scanf("%d",&T);T;T--) 32 { 33 memset(dp,0,sizeof(dp)); 34 int n,num=0,c; 35 scanf("%d",&n); 36 scanf("%d",&col[0]); 37 len[0]=1; 38 for(int i=1;i<n;i++) 39 { 40 scanf("%d",&c); 41 if(col[num]==c) 42 len[num]++; 43 else 44 col[++num]=c,len[num]=1; 45 } 46 printf("Case %d: %d\n",++ca,dfs(0,num,0)); 47 } 48 return 0; 49 }