【题目描述】
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
资源约定:
峰值内存消耗 < 256M
CPU消耗 < 1000ms
【解题思路】
用一个四维数组F[i][j][k][v],表示在点(i,j)小明手中的宝物数为k,最大价值为v,此时的方案数即数组的元素。
而下一个点(右边或下边)的方案数要加上一个节点的方案数,可以理解为继承。
注意:无论小明遇到的宝物价值是多少,他都可以选择不拿;但是要拿,遇到的宝物价值就必须超过手中的宝物价值。
所以分为两种情况:
1.选择不拿,直接加上一个点的方案数
(F[i+1][j][k][v]+=sum)%=MOD; (F[i][j+1][k][v]+=sum)%=MOD;
2.选择拿,判断当前宝物价值,从而选择是否改变手中的宝物数
if(a[i+1][j]>v) (F[i+1][j][k+1][a[i+1][j]]+=sum)%=MOD;
if(a[i][j+1]>v) (F[i][j+1][k+1][a[i][j+1]]+=sum)%=MOD;
还要注意一点,宝物的价值为零不代表没有宝物,因此为了枚举方便,构建图的时候给每个宝物的价值加一。
并且,初始化数组F的时候,要把第一个点拿了一个宝物且宝物价值不为零的方案数置为1.(这样才能AC,虽然还没白为什么=.=)
原创代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#define MOD 1000000007
using namespace std;
int F[110][110][24][24];
int a[110][110];
int n,m,k;
void init()
{
memset(F,0,sizeof(F));
F[1][1][0][0]=1;
if(a[1][1]) F[1][1][1][a[1][1]]=1;
}
void GetTreasure(int r, int c, int maxk, int maxv)
{
for(int i=1; i<=r; i++)
for(int j=1; j<=c; j++)
for(int k=0; k<=maxk; k++)
for(int v=0; v<=maxv; v++)
{
int sum=F[i][j][k][v];
(F[i+1][j][k][v]+=sum)%=MOD;
(F[i][j+1][k][v]+=sum)%=MOD;
if(a[i+1][j]>v)
(F[i+1][j][k+1][a[i+1][j]]+=sum)%=MOD;
if(a[i][j+1]>v)
(F[i][j+1][k+1][a[i][j+1]]+=sum)%=MOD;
}
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
for(int i=1; i<=n; i++)
for(int j=1; j<=m; j++)
{
scanf("%d",&a[i][j]);
a[i][j]++;//宝物价值为0也代表有宝物,为了便于判断,给每个宝物的价值+1
}
init();
GetTreasure(n,m,k,13);
int ans=0;
for(int i=0; i<=13; i++)
(ans+=F[n][m][k][i])%=MOD;
printf("%d\n",ans);
}
参考代码:
#include<iostream>
#include<cstring>
#include<cstdio>
#define MOD %1000000007
#define MAXN 60
using namespace std;
long long r[MAXN][MAXN][15][15];
long long map[MAXN][MAXN];
long long m,n,num;
long long dfs(long long i, long long j, long long k, long long v)
{
long long s=0, t;
if(r[i][j][k][v+1]!=-1)
return r[i][j][k][v+1];
if(i==m&&j==n)
{
if(k==num)
{
r[m][n][k][v+1]=1;
return r[m][n][k][v+1];
}
else if(k==num-1 && map[m][n]>v)
{
r[m][n][k][v+1]=1;
return r[m][n][k][v+1];
}
else
{
r[m][n][k][v+1]=0;
return r[m][n][k][v+1];
}
}
else
{
if(map[i][j]>v)
{
t=map[i][j];
if(i+1<=m)
s=(s+dfs(i+1,j,k+1,t)MOD + dfs(i+1,j,k,v)MOD)MOD;
if(j+1<=n)
s=(s+dfs(i,j+1,k+1,t)MOD + dfs(i,j+1,k,v)MOD)MOD;
}
else
{
if(i+1<=m)
s=(s+dfs(i+1,j,k,v)MOD)MOD;
if(j+1<=n)
s=(s+dfs(i,j+1,k,v)MOD)MOD;
}
r[i][j][k][v+1]=s MOD;
return r[i][j][k][v+1];
}
}
int main()
{
memset(r,-1,sizeof(r));
long long i,j,p,q;
cin>>m>>n>>num;
for(int i=1; i<=m; i++)
for(int j=1; j<=n; j++)
cin>>map[i][j];
dfs(1,1,0,-1);
cout<<r[1][1][0][0]<<endl;
return 0;
}