AcWing 1212. 地宫取宝【动态规划】【第五届蓝桥杯省赛C++A/B/C组,第五届蓝桥杯省赛JAVAB/C组】

---

一、题目链接

AcWing 1212. 地宫取宝

---

二、题目分析

（一）算法标签

动态规划

（二）解题思路

本题其实是摘花生和最长上升子序列的结合版
限制1：只能往右或者下走
限制2：拿宝贝按价值严格单调递增的顺序
限制3：最后拿了k件

1.状态表示

集合： f[i][j][k][c]表示所有从起点走到(i, j)，且已经取了k件物品，且最后一件物品的价值是c的合法方案的集合

属性： 数量

2.状态计算(集合划分)

1. 所有最后一步是从上往下走的走法
   1. 不取：f[i - 1][j][k][c]
   2. 取:
   前提条件：w[i][j] == c 且 c' < c (c'是上一件取的物品的价值)
   f[i - 1][j][k - 1][c']

2. 所有最后一步是从左往右走的走法
   1. 不取: f[i][j - 1][k][c]
   2. 取:
   前提条件：w[i][j] == c 且 c' < c (c'是上一件取的物品的价值)
   f[i][j - 1][k - 1][c']

初始化：
f[1][1][1][w[1][1]] = 1
f[1][1][0][-1] = 1 (这里注意数组下标不能为负, 所以在实际处理时每个w[i][j]都加1)

---

三、AC代码

解法一:

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 55, MOD = 1000000007;

int n, m, k;
int w[N][N];
int f[N][N][13][14];

int main()
{
    cin >> n >> m >> k;
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= m; j ++ )
        {
            cin >> w[i][j];
            w[i][j] ++ ;    // 由于数组下标不能为负数, 所以每一个价值都加上1
        }
    
    //  初始化
    f[1][1][1][w[1][1]] = 1;
    f[1][1][0][0] = 1;
    
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= m; j ++ )
        {
            if (i == 1 && j == 1) continue;
            for (int u = 0; u <= k; u ++ )
                for (int v = 0; v <= 13; v ++ )
                {
                    int &val = f[i][j][u][v];   // val的值改变, f 的值跟着改变
                    // 不取
                    val = (val + f[i - 1][j][u][v]) % MOD;
                    val = (val + f[i][j - 1][u][v]) % MOD;
                    // 取
                    if (u > 0 && w[i][j] == v)  // u > 0 是因为之后的 u - 1 要 >= 0
                    {
                        for (int c = 0; c < v; c ++ )
                        {
                            val = (val + f[i - 1][j][u - 1][c]) % MOD;
                            val = (val + f[i][j - 1][u - 1][c]) % MOD;
                        }
                    }
                }
        }

    int res = 0;
    for (int i = 0; i <= 13; i ++ ) res = (res + f[n][m][k][i]) % MOD;
    cout << res;
    
    return 0;
}

---

四、其它题解

地宫取宝