题目:
问题描述
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
思路:
dfs。用一个变量记录当前最大价值,如果当前所在位置map[i][j]>maxVal,可以选择拿或不拿,当拿到宝物数等于k,并且在最右下角时,路径条数加1。
直接dfs肯定会超时,所以要加上记忆化。
代码:
不加记忆化,会超时的代码。
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
using namespace std;
int map[55][55];
int n, m, k;
const int mod = 1000000007;
long long ans;
void dfs(int i,int j,int cnt,int maxVal)
{
if (cnt > k){
return;
}
if (i == n - 1 && j == m - 1){
if (k == cnt){
ans++;
ans %= mod;
}
else if (k == cnt + 1 && map[n - 1][m - 1] > maxVal){
ans++;
ans %= mod;
}
}
else{
if (map[i][j] > maxVal){
if (i + 1 < n){
dfs(i+1,j,cnt+1,map[i][j]);
dfs(i+1,j,cnt,maxVal);
}
if (j + 1 < m){
dfs(i,j+1,cnt+1,map[i][j]);
dfs(i,j+1,cnt,maxVal);
}
}
else{
if (i + 1 < n){
dfs(i + 1, j, cnt, maxVal);
}
if (j + 1 < m){
dfs(i, j + 1, cnt, maxVal);
}
}
}
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
for (int i = 0; i < n; i++)
{
for (int j = 0; j < m; j++){
scanf("%d",&map[i][j]);
}
}
dfs(0,0,0,-1);
printf("%lld",ans);
return 0;
}
加上记忆化代码 感谢
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;
int map[55][55];
const int mod = 1000000007;
long long d[55][55][15][15];
int m, n, k;
long long dfs(int i, int j, int num, int maxValue){
long long tmp=0;
if (d[i][j][num][maxValue + 1] != -1){
return d[i][j][num][maxValue + 1];
}
if (i == m && j == n){
if (k == num){
d[i][j][num][maxValue + 1] = 1;
return d[i][j][num][maxValue + 1];
}
else if(k==num+1 && map[i][j]>maxValue){
d[i][j][num][maxValue + 1] = 1;
return d[i][j][num][maxValue + 1];
}
else{
d[i][j][num][maxValue + 1] = 0;
return d[i][j][num][maxValue+1];
}
}
else
{
if (map[i][j] > maxValue){
if (i + 1 <= m){
tmp = (tmp+dfs(i+1,j,num+1,map[i][j])%mod+dfs(i+1,j,num,maxValue)%mod)%mod;
}
if (j + 1 <= n){
tmp = (tmp + dfs(i, j + 1, num + 1, map[i][j]) % mod + dfs(i, j + 1, num, maxValue) % mod) % mod;
}
}
else
{
if (i + 1 <= m){
tmp = (tmp+dfs(i+1,j,num,maxValue)%mod)%mod;
}
if (j + 1 <= n){
tmp = (tmp+dfs(i,j+1,num,maxValue)%mod)%mod;
}
}
d[i][j][num][maxValue + 1] = tmp%mod;
return d[i][j][num][maxValue+1];
}
}
int main()
{
scanf("%d%d%d", &m, &n, &k);
for (int i = 1; i<=m; i++){
for (int j = 1; j<=n; j++){
scanf("%d", &map[i][j]);
}
}
memset(d,-1,sizeof(d));
dfs(1, 1, 0, -1);
printf("%lld\n", d[1][1][0][0]);
return 0;
}