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
思路:第一次接触这题,首先想到的是递归出所有可能,但其中存在大量的重复计算,数据太大时会超时(代码如下),后参考别人的代码想到用记忆搜索
代码一(不建议,仅供参考)
#include <stdio.h>
#define MAX 1000000007
#define M 51
int values[M][M]={0};
int n=0,m=0,k=0;
int count=0;
void search(int own,int max,int x,int y);
int main(void)
{
int i=0,j=0;
scanf("%d %d %d",&n,&m,&k);
for(i=0;i<n;i++)
{
for(j=0;j<m;j++)
{
scanf("%d",*(values+i)+j);
}
}
search(0,0,0,0);
printf("%d",count);
return 0;
}
void search(int own,int max,int x,int y) //手里拥有的数量和最大值和坐标
{
if(x>=n||y>=m||own>k) return;
if(x==n-1&&y==m-1&&(own==k||own==k-1&&values[x][y]>max))
{
count++;
count = count%MAX;
return;
}
if(values[x][y]>max)
{
search(own+1,values[x][y],x+1,y);
search(own+1,values[x][y],x,y+1);
}
search(own,max,x+1,y);
search(own,max,x,y+1);
}
代码二(记忆搜索,dfs):
#include <stdio.h>
#include <string.h>
#define M 55
#define N 15
#define MOD %1000000007
int State[M][M][N][N]; //四维数组,n行,m列,拥有宝物数own,手里最大价值max
int values[M][M]={0}; //接收所有宝物的价值
int n=0,m=0,k=0;
int search(int x,int y,int own,int max);
int main(void)
{
memset(State,-1,sizeof(State)); //将所有四维数组值初始化为-1(不能为0)
int i=0,j=0;
scanf("%d %d %d",&n,&m,&k);
for(i=1;i<=n;i++)
{
for(j=1;j<=m;j++)
{
scanf("%d",*(values+i)+j);
}
}
printf("%d",search(1,1,0,-1));
return 0;
}
int search(int x,int y,int own,int max)
{
//如果不为-1,则直接返回此处值(此处已被记忆)
if(State[x][y][own][max+1]!=-1) return State[x][y][own][max+1];
int sum=0,tmp=0;
if(x==n&&y==m) //最终条件
{
if(own==k||own==k-1&&values[x][y]>max)
{
State[x][y][own][max+1] = 1;
}
else State[x][y][own][max+1] = 0;
return State[x][y][own][max+1];
}
else
{
//因为无论此处的宝物价值是否大于手里的max,都可以选择不拿,所以不拿的两种单独拿出来
if(values[x][y]>max)
{
tmp = values[x][y];
if(x+1<=n)
sum = (sum+search(x+1,y,own+1,tmp))MOD;
if(y+1<=m)
sum = (sum+search(x,y+1,own+1,tmp))MOD;
}
if(x+1<=n) sum = (sum+search(x+1,y,own,max))MOD;
if(y+1<=m) sum = (sum+search(x,y+1,own,max))MOD;
State[x][y][own][max+1] = sum;
return sum;
}
}