HDOJ 1078 记忆搜索

第一次，只用了普通的深搜，超时。然后去看了别人的代码，发现可以用记忆搜索。

用二维数组ans保存从某点出发可以积累到的food的最大数量，用深搜从起点开始处理。一个点的ans值一旦算出，之后就可以直接使用，不用再计算。

#include <iostream>
using namespace std;

const int MAP_SIZE = 105;
int n,k;
int map[MAP_SIZE][MAP_SIZE],ans[MAP_SIZE][MAP_SIZE];
int direction_x[5] = { 0,-1,1,0,0 };/* 上下左右 */
int direction_y[5] = { 0,0,0,-1,1 };/* 上下左右 */

int DFS(int x,int y)
{
    int xx,yy;
    int max = 0;
    for (int i = 1;i <= 4;i ++)/* 找四个方向 */
    {
        for (int j = 1;j <= k;j ++)/* 每次之多跑k步 */
        {
            xx = x + direction_x[i] * j;
            yy = y + direction_y[i] * j;
            if (xx < 1 || xx > n || yy < 1 || yy > n)/* 坐标要合法 */
                continue;
            if (map[x][y] >= map[xx][yy])/* 下一个点的食物要比上一个点多 */
                continue;
            int temp;
            if (ans[xx][yy] == -1)/* 如果(xx,yy)点没有计算过，则先计算 */
                temp = DFS(xx,yy);
            else
                temp = ans[xx][yy];/* ans[xx][yy]之前计算过，现在可以直接使用 */
            max = max > temp ? max : temp;
        }
    }
    ans[x][y] = max + map[x][y];
    return ans[x][y];
}

int main ()
{
    while (scanf("%d%d",&n,&k) != -1)
    {
        if (n == -1 && k == -1)
            break;
        memset(ans,-1,sizeof(ans));
        for (int i = 1;i <= n;i ++)
            for (int j = 1;j <= n;j ++)
                scanf("%d",&map[i][j]);
        printf("%d\n",DFS(1,1));
    }
    return 0;
}