初见蓝桥——历年试题 地宫寻宝

蓝桥杯——历年试题 地宫寻宝

问题描述

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

 

这是一道DFS的题目，其实DFS（深度优先搜索）和DP（动态规划）是相通的，代码中所用到的d数组，完全可以作为动态数组来进行动态规划，题目本身难度并不大，在这里我只说明一下d[i][j][p][q]表示处理到(i,j)位置获得p件物品且最大价值为q的方案数。代码如下：

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring> 
#include <queue>
using namespace std;

const int maxn = 100;
const long long INF = 1000000007; //用来取余 
int maze[maxn][maxn];

int d[52][52][14][14]; //行,列,k个数,value
int n, m, k;           //[n,m], k件宝贝
long long ans;         //方案数 

int dfs(int r, int c, int sum, int Max); //当前位置 
void input();
void solve();

void input()
{
    memset(d, -1, sizeof(d));
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++) {
            scanf("%d", &maze[i][j]);
        }
    }
}

int dfs(int r, int c, int sum, int Max)
{
    if (d[r][c][sum][Max + 1] != -1) {      //已经遍历完, 并设置了结果 
        return d[r][c][sum][Max + 1];       //返回结果 
    }

    int t = 0;

    if (r == n - 1 && c == m - 1) {         //到达入口 
                                             
        if (maze[r][c] > Max) {             //可以再拿一个宝物 
            if (sum == k || sum == k - 1)   //如果已经到了 k 或是  k-1，方案++ 
                t++;
        }
        else if (sum == k) {                //不能拿时候,则此时就需要为k, 方案++ 
            t++;
        }
        return d[r][c][sum][Max + 1] = t;   //更新方案数 
    }

    if (r + 1 < n) {
        if (maze[r][c] > Max) {            //可以拿 
            t += dfs(r + 1, c, sum + 1, maze[r][c]);      //选择拿 
            t %= INF;
            t += dfs(r + 1, c, sum, Max);                 //选择不拿 
            t %= INF;
        }
        else {
            t += dfs(r + 1, c, sum, Max);  //不可以拿 
            t %= INF;
        }
    }
    if (c + 1 < m) {
        if (maze[r][c] > Max) {
            t += dfs(r, c + 1, sum + 1, maze[r][c]);
            t %= INF;
            t += dfs(r, c + 1, sum, Max);
            t %= INF;
        }
        else {
            t += dfs(r, c + 1, sum, Max);
            t %= INF;
        }
    }
    d[r][c][sum][Max + 1] = t;         //将方案数 保存在Max + 1 处 
    return d[r][c][sum][Max + 1];      //返回方案数 
}

void solve()
{
    input();
    ans = dfs(0, 0, 0, -1);
    cout << ans << endl;
}

int main()
{
    solve();
    return 0;
}