题目描述
X 国王有一个地宫宝库。是 n x m 个格子的矩阵。每个格子放一件宝贝。每个宝贝贴着价值标签。
地宫的入口在左上角,出口在右下角。
小明被带到地宫的入口,国王要求他只能向右或向下行走。
走过某个格子时,如果那个格子中的宝贝价值比小明手中任意宝贝价值都大,小明就可以拿起它(当然,也可以不拿)。
当小明走到出口时,如果他手中的宝贝恰好是k件,则这些宝贝就可以送给小明。
请你帮小明算一算,在给定的局面下,他有多少种不同的行动方案能获得这k件宝贝。
输入
输入一行3个整数,用空格分开:n m k (1< =n,m< =50, 1< =k< =12)
接下来有 n 行数据,每行有 m 个整数 Ci (0< =Ci< =12)代表这个格子上的宝物的价值
输出
要求输出一个整数,表示正好取k个宝贝的行动方案数。该数字可能很大,输出它对 1000000007 取模的结果。
样例输入
2 3 2 1 2 3 2 1 5
样例输出
14
WA和超时了一个中午,总结一下记忆化搜索,其实思路已经到位了,就是学到了很多边界的处理,先从开空间说起,有多少个参数就开多少维空间,memo数组的初始值设为-1,而不是0,不然记忆搜索会退化导致超时,另外一个要注意的就是数组的值范围为0-12,所以你穿入的max不能为0,应该为-1,但是开的空间下标总不能为负数把,所以可以让所有数组里面的值+1就等价于max改为-1了。
import java.util.Scanner;
public class 地宫取宝 {
static int n,m,k;
static int[][] a = new int[55][55];
static int[][][][] memo = new int[55][55][13][13];
static int mod = 1000000007;
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
n = in.nextInt();
m = in.nextInt();
k = in.nextInt();
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
a[i][j] = 1+in.nextInt();
for(int i=0;i<=54;i++)
for(int j=0;j<=54;j++)
for(int ii=0;ii<=12;ii++)
for(int jj=0;jj<=12;jj++)
memo[i][j][ii][jj]=-1;
System.out.println(dfs(1,1,0,0));
}
static int dfs(int x,int y,int t,int max) {
if(memo[x][y][t][max]!=-1)
return memo[x][y][t][max];
memo[x][y][t][max]=0;
if(x==n && y==m && t==k)
memo[x][y][t][max] = 1;
if(a[x][y]>max && t<k) {
memo[x][y][t][max] += dfs(x,y,t+1,a[x][y]);
memo[x][y][t][max]%=mod;
}
if(x<n) {
memo[x][y][t][max] += dfs(x+1,y,t,max);
memo[x][y][t][max]%=mod;
}
if(y<m) {
memo[x][y][t][max] += dfs(x,y+1,t,max);
memo[x][y][t][max]%=mod;
}
return memo[x][y][t][max];
}
}
这几个小时花的挺值的,对记忆的理解更进一步了