地宫取宝——第五届蓝桥杯省赛C语言A组第8题

【题目描述】

    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;
}