题目描述
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 3 2
1 2 3
2 1 5
样例输出
14
#include<iostream>
using namespace std;
#define M 1000000007
int m,n,k;
int arr[51][51];
long long s=0;
//第一次调用参数为 1,1,0,0
// a,b 当前坐标 cur当前拿到的宝贝数目 max当前拿到的宝贝的最大价值
void dfs(int a,int b,int cur,int max){
if(a==m&&b==n){
if(cur==k||(cur==k-1&&max<arr[a][b])){
s++;
return;
}
else
return;
}
//向下走
if(a+1<=m){
if(max<arr[a][b])
dfs(a+1,b,cur+1,arr[a][b]);
dfs(a+1,b,cur,max);
}
//向右走
if(b+1<=n){
if(max<arr[a][b])
dfs(a,b+1,cur+1,arr[a][b]);
dfs(a,b+1,cur,max);
}
}
int main(){
cin>>m>>n>>k;
for(int i=1;i<=m;i++){
for(int j=1;j<=n;j++){
cin>>arr[i][j];
}
}
dfs(1,1,0,0);
cout<<s%M<<endl;
return 0;
}
结合动态规划,降低复杂度
动态规划状态转移方程:
#include<iostream>
using namespace std;
#define M 1000000007
#include<cstring>
int n,m,k;
int arr[51][51];
int dp[50][50][15][15];
int move[2][2]={{0,1},{1,0}};
int dfs(int x,int y,int cur,int max){
//如果当前这一步走过了,直接返回值就好了
if (dp[x][y][cur][max+1]!=-1)
return dp[x][y][cur][max+1];
if(x==n-1&&y==m-1){//如果走到了右下角
if(cur==k||(cur==k-1&&max<arr[x][y])){//不取,或者取 右下角宝物,刚好满足宝物数为K , 就加1
return dp[x][y][cur][max+1]=1;
}else{
return dp[x][y][cur][max+1]=0;
}
}
long long s=0;
for(int i=0;i<2;i++){//i=0时,向右走,i=1时,向下走
int x1=x+move[i][0];
int y1=y+move[i][1];
if(x1<n&&y1<m){//满足在数组范围内
if(max<arr[x][y]){//当前的宝物最大价值小于所在位置的宝物价值
s+=dfs(x1,y1,cur+1,arr[x][y]);//拿宝物
}
s+=dfs(x1,y1,cur,max);//不拿宝物
}
}
return dp[x][y][cur][max+1]=s%M;
}
int main(){
cin>>n>>m>>k;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
cin>>arr[i][j];
}
}
memset(dp,-1,sizeof(dp));//数组元素初始化为-1
dfs(0,0,0,-1);//dfs(x,y,cur,max) 当前位置坐标, 当前宝物数目, 当前宝物最大值 ,将宝物最大值设为-1,因为宝物的价值可能会为0
cout<<dp[0][0][0][0]<<endl;
return 0;
}