蓝桥杯地宫寻宝【dp】

最新推荐文章于 2022-03-30 12:14:22 发布

笑着走完自己的路

最新推荐文章于 2022-03-30 12:14:22 发布

阅读量1k

点赞数

分类专栏：我就是DP

本文链接：https://blog.csdn.net/chenzhenyu123456/article/details/50931978

版权

我就是DP 专栏收录该内容

91 篇文章 0 订阅

订阅专栏

本文介绍了一道关于地宫寻宝的算法题，玩家从左上角开始，只能向右或向下移动，目标是选择价值最大的k件宝贝。题目要求计算在给定条件下，达到目标的行动方案数。输入包括地宫的大小n和m以及要获取的宝贝数k，输出是满足条件的方案数模1000000007的结果。样例给出了两种不同情况的输入和输出。解决方案中提到可以使用动态规划（dp）来处理此问题。

摘要由CSDN通过智能技术生成

问题描述
　　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

宝物价值可能为0，感觉会出问题，先全加一。
$dp[i][j][p][q]$ 表示处理到 $(i, j)$ 位置获得 $p$ 件物品且最大价值为 $q$ 的方案数。

$AC$ 代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#define PI acos(-1.0)
#define CLR(a, b) memset(a, (b), sizeof(a))
#define fi first
#define se second
#define ll o<<1
#define rr o<<1|1
using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
const int MAXN = 1e5 + 10;
const int pN = 1e6;// <= 10^7
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
void add(LL &x, LL y) { x += y; x %= MOD; }
LL dp[52][52][15][15];
int Map[52][52];
int main()
{
    int n, m, k; scanf("%d%d%d", &n, &m, &k);
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= m; j++) {
            scanf("%d", &Map[i][j]);
            Map[i][j]++;
        }
    } CLR(dp, 0);
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= m; j++) {
            if(i == 1 && j == 1) {
                dp[i][j][0][0] = 1;
                dp[i][j][1][Map[i][j]] = 1;
            }
            else if(i == 1) {
                for(int p = 0; p <= k; p++) {
                    for(int q = 0; q <= 12; q++) {
                        if(dp[i][j-1][p][q]) {
                            add(dp[i][j][p][q], dp[i][j-1][p][q]);
                            if(q < Map[i][j]) {
                                add(dp[i][j][p+1][Map[i][j]], dp[i][j-1][p][q]);
                            }
                        }
                    }
                }
            }
            else if(j == 1) {
                for(int p = 0; p <= k; p++) {
                    for(int q = 0; q <= 12; q++) {
                        if(dp[i-1][j][p][q]) {
                            add(dp[i][j][p][q], dp[i-1][j][p][q]);
                            if(q < Map[i][j]) {
                                add(dp[i][j][p+1][Map[i][j]], dp[i-1][j][p][q]);
                            }
                        }
                    }
                }
            }
            else {
                for(int p = 0; p <= k; p++) {
                    for(int q = 0; q <= 12; q++) {
                        if(dp[i][j-1][p][q]) {
                            add(dp[i][j][p][q], dp[i][j-1][p][q]);
                            if(q < Map[i][j]) {
                                add(dp[i][j][p+1][Map[i][j]], dp[i][j-1][p][q]);
                            }
                        }
                        if(dp[i-1][j][p][q]) {
                            add(dp[i][j][p][q], dp[i-1][j][p][q]);
                            if(q < Map[i][j]) {
                                add(dp[i][j][p+1][Map[i][j]], dp[i-1][j][p][q]);
                            }
                        }
                    }
                }
            }
        }
    }
    LL ans = 0;
    for(int i = 0; i <= 12; i++) {
        add(ans, dp[n][m][k][i]);
    }
    printf("%lld\n", ans);
    return 0;
}