标题:地宫取宝
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 2 2
1 2
2 1
程序应该输出:
2
再例如,输入:
2 3 2
1 2 3
2 1 5
程序应该输出:
14
资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 2000ms
请严格按要求输出,不要画蛇添足地打印类似:“请您输入…” 的多余内容。
所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
注意:不要使用package语句。不要使用jdk1.7及以上版本的特性。
注意:主类的名字必须是:Main,否则按无效代码处理。
public class 地宫取宝记忆 {
static int n, m, k;
static int[][] grid;
static int mod = 1000000007;
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
n = scanner.nextInt();
m = scanner.nextInt();
k = scanner.nextInt();
grid = new int[n][m];
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
grid[i][j] = scanner.nextInt();
}
}
// y x (-1~12)max+1 k(0-12)
cache = new int[n][m][13][k + 2];
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
for (int k = 0; k < 13; k++) {
for (int l = 0; l < 地宫取宝记忆.k + 1; l++) { // 0 1 2 3 ... k
cache[i][j][k][l] = -1;
}
}
}
}
System.out.println(f(0, 0, -1, 0));
}
static int ans = 0;
static int[][][][] cache;
// max 当前最大 kk 几件
// 记忆递归 4维 ny,mx,max,
// max 是当前手里宝贝的最大价值 因为还有一种情况是手里没有任何宝贝
// 没有任何宝贝记做-1 数组下表从0开始 所以-1给+1位偏移0上
public static int f(int ny, int mx, int max, int kk) {
int ra;
if ((ra = cache[ny][mx][max + 1][kk]) != -1)
return ra;
if (ny == n - 1 && mx == m - 1) {
if (kk == k || (kk == k - 1 && max < grid[ny][mx])) {
ans++;
if (ans > mod) {
ans %= mod;
}
return 1;
}
return 0;
}
if (kk > k) // 大剪
return 0;
int value = grid[ny][mx];
int ans = 0;
if (value > max) { // 拿
if (checkPosition(ny + 1, mx))
ans += cache[ny + 1][mx][value + 1][kk + 1] = f(ny + 1, mx, value, kk + 1) % mod;
if (checkPosition(ny, mx + 1))
ans += cache[ny][mx + 1][value + 1][kk + 1] = f(ny, mx + 1, value, kk + 1) % mod;
}
if (checkPosition(ny + 1, mx))
ans += cache[ny + 1][mx][max + 1][kk] = f(ny + 1, mx, max, kk) % mod;
if (checkPosition(ny, mx + 1))
ans += cache[ny][mx + 1][max + 1][kk] = f(ny, mx + 1, max, kk) % mod;
return ans % mod;
}
public static boolean checkPosition(int y, int x) {
return y >= 0 && y < n && x >= 0 && x < m;
}
}