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);
    }
}