第五届蓝桥杯C/C++本科b组 第九题 地宫取宝

问题描述

　　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

看了这题，第一反应就是dp，但考场上一时想不出dp方程⊙﹏⊙b汗只能深搜， 虽然肯定会超时。。。

回来若干天后，发现可以记忆化搜索，即可ac

代码如下:

#include<stdio.h>
#include<string.h>

#define M 1000000007

int g[60][60];
int n,m,k,ans;
int d[51][51][13][13];

int dfs(const int &x,const int &y,const int &max,const int &cnt)
{

	if (d[x][y][max+1][cnt]!=-1)
		{
			return d[x][y][max+1][cnt];
		}

	
    if (n+m-x-y+1+cnt<cnt) return 0;//一个小小的剪枝，不用也可以过
	
	int t=0;

    if (x==n-1&&y==m-1)
    {
		if ((cnt==k)||(cnt==k-1&&max<g[x][y]))
			{
				t++;
			}
		return d[x][y][max+1][cnt]=t;
    }

    if (x+1<n)
	{
        t=(t+dfs(x+1,y,max,cnt))%M;
	}

	if (y+1<m)
    {    
		t=(t+dfs(x,y+1,max,cnt))%M;
	}

    if (g[x][y]>max)
    {
        if (x+1<n) 
            t=(t+dfs(x+1,y,g[x][y],cnt+1))%M;

        if (y+1<m)
            t=(t+dfs(x,y+1,g[x][y],cnt+1))%M;
    }

	d[x][y][max+1][cnt]=t;
	return t;
}
int main()
{
    
    scanf("%d%d%d",&n,&m,&k);
    int i,j,p;
    for (i=0;i<n;++i)
        for (j=0;j<m;++j)
            scanf("%d",&g[i][j]);    

	memset(d,-1,sizeof(d));
	             
    printf("%d\n",dfs(0,0,-1,0));//前两个参数是当前坐标，第三个参数是已选数中的最大值，初始必须是一个负数，第四个是已选数的个数
    //while (1);
	return 0;
}