地宫寻宝
问题描述
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
思路
本题与小明登楼梯的问题相似,类似于小明在登楼梯的过程中添加了限制条件。
小明登楼梯
回到本题,分析题干给出的限制条件。(x,y)表示当前坐标,max为持有物品的最大价值,cur为当前坐标物品的价值,count为持有物品数目。
- 只能向下或者向右走(与机器人走方格的问题类似)((x+1,y)或(x,y+1))
- 对于每件物品,符合条件则可以选择带走或不带走该物品(max=cur,count++)
所以,只要在出口时,满足下列两个情景即可算一种可行的方案。
- count = k
- count = k - 1 && cur > max(表示在出口时,恰好差一件物品,而此时又恰好可以拿走放置在出口的物品)
注意此方法只能通过一半的测试数据
import java.util.Scanner;
public class 地宫寻宝 {
static int k;
static int[][] data;
static int ans;
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner sc = new Scanner(System.in);
while (sc.hasNext()) {
ans = 0;
int n = sc.nextInt();
int m = sc.nextInt();
k = sc.nextInt();
data = new int[n][m];
for (int i = 0; i < data.length; i++) {
for (int j = 0; j < data[i].length; j++) {
data[i][j] = sc.nextInt();
}
}
dfs(0, 0, -1, 0);
System.out.println(ans);
}
}
private static void dfs(int x, int y, int max, int count) {
if (x == data.length || y == data[0].length) {
return;
}
int cur = data[x][y];
if (x == data.length - 1 && y == data[0].length - 1) {// 此时位于出口决策
if (count == k || (count == k - 1 && cur > max)) {
ans++;
if (ans > 1000000007) {
ans %= 1000000007;
}
}
return;
}
if (cur > max) {
dfs(x, y + 1, cur, count + 1);
dfs(x + 1, y, cur, count + 1);
}
// 对于价值较小,或者价值大但不取这个物品的情况如下
dfs(x, y + 1, max, count);
dfs(x + 1, y, max, count);
}
}