hdu1078(记忆化搜索)

题目链接FatMouse and Cheese

一个网格上有很多cheese，老师偷cheese吃，每次四邻域最多走k步，下一步的cheese数必须必当前位置多，求能吃到的最多cheese

记忆化搜索

如图，从第一行的数开始，每次可以往左下或右下走一格，直到走到最下行，把沿途经过的数全部加起来。 如何走才能使得这个和尽量大？

d(i, j)为从格子(i, j)出发时能得到的最大和（包括格子(i, j)本身的值）状态转移方程为

dp[i,j]=max(dp[i+1,j],dp[i+1,j+1])

代码

int solve(int i, int j){
   return a[i][j] + (i == n ? 0 : max(solve(i＋1,j),solve(i＋1,j＋1)));
}

如图，上式的转移方程走过了很多重复的路线，

1、递推法避免重复计算

int i, j;
for(j = 1; j ＜= n; j＋＋) d[n][j] = a[n][j];
     for(i = n－1; i ＞= 1; i——)
         for(j = 1; j ＜= i; j＋＋)
              d[i][j] = a[i][j] ＋ max(d[i＋1][j],d[i＋1][j＋1]);

2、记忆化搜索面重复计算

int solve(int i, int j){
    if(d[i][j] ＞= 0) return d[i][j];//通过判断是否d[i][j]≥0得知它是否已经被计算过
         return d[i][j] = a[i][j] ＋ (i == n ? 0 : max(solve(i＋1,j),solve(i＋1,j＋1)));

回到本题

状态方程为

d[i][j] = a[i][j] ＋ max(dfs(x+k*dx[i],y+k*dy[i]))

代码

#include <iostream>
#include <stdio.h>
#include <memory.h>
#include <algorithm>
using namespace std;
const int maxn=105;
int n,k;
int g[maxn][maxn];
int dp[maxn][maxn],vis[maxn][maxn];
int dx[]={-1,1,0,0};
int dy[]={0,0,-1,1};
bool check(int x,int y)
{
    if(x<0||x>n-1||y<0||y>n-1)return false;
    return true;
}

int dfs(int x,int y)
{
    //cout<<x<<","<<y<<endl;
    int ans=0;
    if(dp[x][y])return dp[x][y];//记忆化搜索部分
    for(int i=0;i<4;i++){
        for(int j=1;j<=k;j++){
            int nx=x+j*dx[i];
            int ny=y+j*dy[i];
            //cout<<x<<","<<y<<endl;
            //cout<<nx<<","<<ny<<endl;
            if(check(nx,ny)){
                if(g[x][y]<g[nx][ny])
                    ans=max(ans,dfs(nx,ny));
            }
        }
        dp[x][y]=ans+g[x][y];
    }
    return dp[x][y];
}

int main()
{
    //freopen("in.txt","r",stdin);
    while(scanf("%d%d",&n,&k)!=EOF&&n!=-1&&k!=-1){
        memset(dp,0,sizeof(dp));
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                scanf("%d",&g[i][j]);
            }
        }
        printf("%d\n",dfs(0,0));
    }
    return 0;
}