依然在DP中挣扎的我。。今天听队友仔细的讲了记忆化搜索与DP的差别 才恍然大悟,01背包之所以可以写成循环的形式,是因为在求解下一状态的最优解时,前面的状态已经解决,但记忆化搜索解决的问题类型,是子问题与总问题有相类似的结构,因此需要层层递归至最底层才能求解。
题目链接:UVA - 10559
题意就是消除方块嘛,得分是个数的平方。。然后这个状态。。是i-j右边拼上k个能得到的最大得分,刚开始我是循环写的,样例过了但是wa了,后来改的记忆化搜索,预处理每个方块向左延伸的长度,每个状态有两种决策,直接消去,和 右边拼上几块再消去。
#include <map>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 205;
int T,n,a[maxn],dp[maxn][maxn][maxn],p[maxn],cases=0;
int solve(int i,int j,int k) {
if(dp[i][j][k]!=-1) return dp[i][j][k];
int& ans=dp[i][j][k];
int q;
if(p[j]==i-1) p[j]++;
if(p[j]==i) return ans=(j-p[j]+1+k)*(j-p[j]+1+k);
ans=solve(i,p[j]-1,0)+(j-p[j]+1+k)*(j-p[j]+1+k);
for(int q=p[j]-1;q>=i;q--) if(a[q]==a[j]&&a[q+1]!=a[j])
ans=max(ans,solve(i,q,k+j-p[j]+1)+solve(q+1,p[j]-1,0));
return ans;
}
int main() {
scanf("%d",&T);
while(T--){
int ans=0;
memset(dp,-1,sizeof(dp));
scanf("%d",&n);
int color=-1,last;
for(int i=1;i<=n;i++){
scanf("%d",a+i);
if(a[i]!=color){
color=a[i];
last=i;
p[i]=i;
continue;
}
p[i]=last;
}
// for(int j=1;j<=n;j++){
// for(int i=1;i<=j;i++){
// for(int k=0;k<=n-j;k++){
// dp[i][j][k]=dp[i][p[j]-1][0]+(j-p[j]+k+1)*(j-p[j]+k+1);
// for(int q=1;q<p[j];q++){
// if(a[q]==a[j]&&a[q+1]!=a[j])
// dp[i][j][k]=max(dp[i][j][k],dp[q+1][p[j]-1][0]+dp[i][q][j-p[j]+k+1]);
// }
// ans=max(ans,dp[i][j][k]);
// }
// }
// }
printf("Case %d: %d\n",++cases,solve(1,n,0));
}
return 0;
}