HDU 1078 记忆化搜索

题目链接

题意是老鼠一开始在（0，0）点，每个点都有奶酪数目，如果老鼠走到哪里的话就获得所有奶酪，老鼠只能水平走和垂直走，问怎么走才能获得最大奶酪数。输入N和K，N是地图大小，K是老鼠能走多远。

很简单的一道题，注意超时，记住用记忆化，貌似就因为要记忆化才被HDU归类进DP的。

下面是代码

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <iostream>
using namespace std;

const int dx[] = {1,-1,0,0};
const int dy[] = {0,0,1,-1};

int mp[101][101];
int vis[101][101];
int dp[101][101];

int n,k,ans;


int dfs(int x,int y)
{
    if(dp[x][y]!=0) return dp[x][y]; //一开始没记忆化TLE了
    int sum = 0;
    sum += mp[x][y];
    int t = 0;
    for(int i=0;i<4;i++){
        for(int j=1;j<=k;j++){
            int xn = x + j*dx[i];
            int yn = y + j*dy[i];
            if(xn < 0 || yn < 0 || xn >= n ||yn >= n) continue;
            if( mp[xn][yn] <= mp[x][y] || vis[xn][yn] ) continue;
            vis[xn][yn] = 1;
            t = max(t,dfs(xn,yn));
            vis[xn][yn] = 0;
        }
    }
    sum += t;
    return dp[x][y] =sum;

}


int main()
{
    while(~scanf("%d%d",&n,&k)){
        if(n==-1 && k==-1) break;
        memset(mp,0,sizeof(mp));
        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",&mp[i][j]);


        ans = dfs(0,0);
        printf("%d\n",ans);
    }
}