http://acm.hdu.edu.cn/showproblem.php?pid=1078
题目大意:输入n,有一个n*n的方格,每个方格里面有0-100个奶酪,老鼠起始在(0,0)方格,每次最多水平或竖直走k个方格,每到一个方格就拿到方格里的所有奶酪。每次只能到比当前方格奶酪数更多的方格中去。求当老鼠不能移动的时候,拿到的最多奶酪数。
题目用到了记忆化搜索,我用的记忆化搜索方法和网上给出的答案比稍微麻烦一些,但好歹也是自己想出来的,就记下来留个纪念吧。
用dp[i][j]表示最终停留在(i,j)的最多奶酪数。则dp[i][j]的值为上一步所有可能的位置的最多奶酪数+(i,j)位置的奶酪数。
比如,当k=1的时候,我们求dp[1][1],cheese[1][1]表示(1,1)处的奶酪数,从(1,0),(0,1),(2,1),(1,2)处有可能下一步到达(1,1)。设(1,0)和(2,1)处的奶酪比(1,1)处的少,则dp[1][1]= max( dp[1][0]+cheese[1][1],dp[2][1]+cheese[1][1] )。
另外需要标识某个位置不可能到达。例如,一个3×3的方格:
3 4 5
6 1 10
11 12 13
(1,1)位置的奶酪数为1,属于不可达的位置。
#include<iostream>
#include<algorithm>
using namespace std;
int n,k;
int cheese[104][104];
int dp[104][104];//从(0,0)出发,最终位置停在i,j最多奶酪数
const int unaccessible=-2;
void dpInit(){
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
dp[i][j]=-1;
}
}
dp[0][0]=cheese[0][0];
}
//计算dp[i][j]
int caldp(int i,int j){
if(dp[i][j]!=-1) return dp[i][j];
int tmp=-1;
for(int m=-1*k;m<=k;m++){
int ret;
if( (i+m>=0) && (i+m)<n && cheese[i][j]>cheese[i+m][j]){
ret=caldp(i+m,j);
if(ret!=unaccessible) tmp=max(tmp,ret+cheese[i][j]);
}
if( (j+m>=0) && (j+m)<n && cheese[i][j]>cheese[i][j+m]){
ret=caldp(i,j+m);
if(ret!=unaccessible) tmp=max(tmp,ret+cheese[i][j]);
}
}
dp[i][j]=tmp;
if(dp[i][j]==-1) dp[i][j]=unaccessible;
return dp[i][j];
}
int main(){
int max_blocks;
while(cin>>n>>k){
if(n==-1 && k==-1) break;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
cin>>cheese[i][j];
}
}
dpInit();
max_blocks=-1;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
max_blocks=max(caldp(i,j),max_blocks);
}
}
/*
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
cout<<"dp["<<i<<"]["<<j<<"]="<<dp[i][j]<<endl;
}
}
*/
cout<<max_blocks<<endl;
}
}