题目描述
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
思路
很明显是一道dp题,看数据很大必须要进行记忆化搜索.
安排一个四维数组dp[x][y][sum][max]
意思是在走到x,y点的时候,手里已经有sum件宝物,最大价值为max
到达终点的时候检查一下sum是否等于k即可
需要注意的是,递归起点不能从x=0,y=0,sum=0,max=0开始,要从max=-1开始。因为一开始什么也没取并不存在max,但是数组又不存在下标-1,只要在以后的递归中把max+1即可,看代码
#include<bits/stdc++.h>
#define mod 1000000007
using namespace std;
int dp[50][50][13][13];//比如dp[3][4][5][6]就是在(3,4)这个点时手里拥有5个宝物,最大价值为6
int mp[50][50];
int n,m,k;
int dfs(int x,int y,int sum,int max)
{
if(dp[x][y][sum][max+1]!=-1)//宝物的价值区间为[0-12],应初始化为-1,因为一开始啥也不取时不存在max
return dp[x][y][sum][max+1];//而max要作为数组下标,显然最小为0,故初始化为-1,然后都+1就解决了,其实max+1就是当前状态
if(x==n&&y==m)//如果到达终点
{
if(sum==k||(sum==k-1&&max<mp[x][y]))//如果手中的宝物已经有k个则选择不拿终点的宝物 或者 手中的宝物有k-1个
return dp[x][y][sum][max+1]=1;//而此时终点的宝物价值大于max,便拿起终点的宝物(对终点宝物的拿与不拿)
else
return dp[x][y][sum][max+1]=0;//不符合就表示该方案不可行,返回0
}
long long cnt = 0;
if(x+1<=n)//如果可以向下走
{
if(max<mp[x][y])//当前格子大于手中宝物的最大价值,即可以拿起时可以选择拿与不拿
cnt=(cnt+dfs(x+1,y,sum+1,mp[x][y]))%mod;
cnt=(cnt+dfs(x+1,y,sum,max)%mod;
}
if(y+1<=m)//如果可以向右走
{
if(max<mp[x][y])//当前格子大于手中宝物的最大价值,即可以拿起时可以选择拿与不拿
cnt=(cnt+dfs(x,y+1,sum+1,mp[x][y]))%mod;
cnt=(cnt+dfs(x,y+1,sum,max)%mod;
}
return dp[x][y][sum][max+1]=cnt;
}
int main()
{
cin >> n >> m >> k;
memset(dp,-1,sizeof(dp));
for(int i = 1;i <= n;i++)
for(int j = 1;j <= m;j++)
cin >> mp[i][j];
dfs(1,1,0,-1);
cout << dp[1][1][0][0];
}