历届试题 地宫取宝
时间限制:1.0s 内存限制:256.0MB
问题描述
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
思路:这道题明显可以用动态规划解决。用f[i][j][k][l]表示小明从第i行第j列出发,欲获得k件宝物,且之前所拿
宝物最大价值为l,满足上述前提下到出口的行动方案数。注意,这里针对输入的格子宝物价值要进行处理,
使得之前没拿过宝物与之前拿到宝物最大价值为0这两种情况区分开来。我一开始没处理,结果就栽在有
宝物价值为0的数据上面去了,只拿20分,后来实在不解看了测试数据才知道~~~~~~我后来直接将各位置
上宝物价值加1。然后首先不拿有两种走法,方案数为f[i+1][j][k][l] + f[i][j+1][k][l] (往下走与往右走两种情况)
而若k>0(即还要拿宝物)且当前宝物价值val>l,则可拿走当前位置上宝物,此时多了两种方案,该两种方案
方案数和为f[i+1][j][k-1][val] +f[i][j+1][k-1][val]。以上所有方案数和即为f[i][j][k][l]。
然后就是边界值的问题。首先是走出的情况(例如i>n),此时方案数为0,即数组默认值,所以可不用进行操作。
接下来就是出口的情况。对于f[n][m][k][l],若k=0,则值都赋为1(此时显然就一种方案,即不取宝物直接走出)。
而k=1时,说明小明想要拿位于出口位置的宝物,设其价值为value。若l<value,则可拿,方案数为1。否则,
小明想拿但拿不了,方案数为0。
最后所求结果即f[1][1][k][0](这里的k指代题目上所给定的k,l=0表示此前未拿宝物),然后注意动态规划时还要
取模的问题即可(为保险起见,f数值用long型存储,防止数值溢出)。
代码如下:
import java.util.Scanner;
public class Main {
static int n, m, k;
static int c[][];
static long f[][][][];
static long num = 1000000007;
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner reader = new Scanner(System.in);
n = reader.nextInt();
m = reader.nextInt();
k = reader.nextInt();
c = new int[n + 1][m + 1];
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++) {
c[i][j] = reader.nextInt();
c[i][j]++;
}
f = new long[n + 2][m + 2][k + 1][14];
for (int v = 0; v <= 13; v++)
f[n][m][0][v] = 1;
for (int v = 0; v < c[n][m]; v++) {
f[n][m][1][v] = 1;
}
long num1, num2, num3, num4;
for (int i = n; i >= 1; i--)
for (int j = m; j >= 1; j--) {
if (!((i == n) && (j == m))) {
for (int l = 0; l <= k; l++) {
for (int v = 0; v <= 13; v++) {
num1 = f[i][j + 1][l][v];
num2 = f[i + 1][j][l][v];
if ((l == 0) || (v >= c[i][j])) {
num3 = 0;
num4 = 0;
} else {
num3 = f[i][j + 1][l - 1][c[i][j]];
num4 = f[i + 1][j][l - 1][c[i][j]];
}
f[i][j][l][v] = (((num1 + num2) % num + num3) % num + num4)
% num;
}
}
}
}
System.out.println(f[1][1][k][0]);
}
}