[蓝桥杯 2014 省 AB] 地宫取宝
题目描述
X 国王有一个地宫宝库。是 n × m n \times m n×m 个格子的矩阵。每个格子放一件宝贝。每个宝贝贴着价值标签。
地宫的入口在左上角,出口在右下角。
小明被带到地宫的入口,国王要求他只能向右或向下行走。
走过某个格子时,如果那个格子中的宝贝价值比小明手中任意宝贝价值都大,小明就可以拿起它(当然,也可以不拿)。
当小明走到出口时,如果他手中的宝贝恰好是 k k k 件,则这些宝贝就可以送给小明。
请你帮小明算一算,在给定的局面下,他有多少种不同的行动方案能获得这 k k k 件宝贝。
输入格式
输入一行 3 3 3 个整数,用空格分开: n n n, m m m, k ( 1 ≤ n , m ≤ 50 , 1 ≤ k ≤ 12 ) k(1 \le n,m \le 50,1 \le k \le 12) k(1≤n,m≤50,1≤k≤12)。
接下来有 n n n 行数据,每行有 m m m 个整数 C i ( 0 ≤ C i ≤ 12 ) C_i(0 \le C_i \le 12) Ci(0≤Ci≤12) 代表这个格子上的宝物的价值。
输出格式
要求输出一个整数,表示正好取 k k k 个宝贝的行动方案数。该数字可能很大,输出它对 1000000007 ( 1 0 9 + 7 ) 1000000007(10^9+7) 1000000007(109+7) 取模的结果。
样例 #1
样例输入 #1
2 2 2
1 2
2 1
样例输出 #1
2
样例 #2
样例输入 #2
2 3 2
1 2 3
2 1 5
样例输出 #2
14
提示
时限 1 秒, 256M。蓝桥杯 2014 年第五届省赛
代码
#include<iostream>
#include<cstring>
using namespace std;
int n, m, k;
const int N = 55, MOD = 1000000007;
int dp[N][N][15][20], w[N][N];
int main() {
cin >> n >> m >> k;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
cin >> w[i][j];
w[i][j]++;
}
}
dp[1][1][1][w[1][1]] = 1;
dp[1][1][0][0] = 1;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
for (int u = 0; u <= k; u++)
for (int v = 0; v <= 13; v++) {
int& val = dp[i][j][u][v];
val = (val + dp[i - 1][j][u][v]) % MOD;
val = (val + dp[i][j - 1][u][v]) % MOD;
if (u > 0 && v == w[i][j]) {
for (int c = 0; c < v; c++) {
val = (val + dp[i][j - 1][u - 1][c]) % MOD;
val = (val + dp[i - 1][j][u - 1][c]) % MOD;
}
}
}
int res = 0;
for (int c = 0; c <= 13; c++) {
res = (res + dp[n][m][k][c]) % MOD;
}
cout << res;
return 0;
}