问题描述
X 国王有一个地宫宝库。是 n x m 个格子的矩阵。每个格子放一件宝贝。每个宝贝贴着价值标签。
地宫的入口在左上角,出口在右下角。
小明被带到地宫的入口,国王要求他只能向右或向下行走。
走过某个格子时,如果那个格子中的宝贝价值比小明手中任意宝贝价值都大,小明就可以拿起它(当然,也可以不拿)。
当小明走到出口时,如果他手中的宝贝恰好是k件,则这些宝贝就可以送给小明。
请你帮小明算一算,在给定的局面下,他有多少种不同的行动方案能获得这k件宝贝。
输入格式
输入一行3个整数,用空格分开:n m k (1<=n,m<=50, 1<=k<=12)
接下来有 n 行数据,每行有 m 个整数 Ci (0<=Ci<=12)代表这个格子上的宝物的价值
输出格式
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 取模的结果。
解:直接递归走的话会超时,原因是重复计算太多,可以使用记忆化DFS,将计算结果存储在数组中,代码:
#include <iostream>
#include <cstring>
using namespace std;
#define N 1000000007
int temp[50][50];
int dfs[50][50][15][15];
int n, m, k;
int step(int nn, int mm, int now, int lar){
if (dfs[nn][mm][now][lar] != -1) {
return dfs[nn][mm][now][lar];
}
if (nn == n-1 && mm == m-1) {
if(temp[nn][mm] > lar){
if(now == k || now == k-1) return dfs[nn][mm][now][lar] = 1;
}
else if(now == k){
return dfs[nn][mm][now][lar] = 1;
} else {
return dfs[nn][mm][now][lar] = 0;
}
}
int res = 0;
if (nn < n-1) {
if (temp[nn][mm] > lar) {
res += step(nn+1, mm, now, lar);
res %= N;
res += step(nn+1, mm, now+1, temp[nn][mm]);
res %= N;
} else {
res += step(nn+1, mm, now, lar);
res %= N;
}
}
if (mm < m-1) {
if (temp[nn][mm] > lar) {
res += step(nn, mm+1, now, lar);
res %= N;
res += step(nn, mm+1, now+1, temp[nn][mm]);
res %= N;
} else {
res += step(nn, mm+1, now, lar);
res %= N;
}
}
return dfs[nn][mm][now][lar] = res;
}
int main() {
memset(dfs,-1,sizeof(dfs));
cin >> n >> m >> k;
for (int i = 0; i < n; i++)
{
for (int j = 0; j < m; j++)
{
cin >> temp[i][j];
temp[i][j]++;
}
}
cout << step(0, 0, 0, 0) << endl;
}