Acwing-蓝桥杯-1212.地宫取宝

题目内容

X 国王有一个地宫宝库，是 n×m个格子的矩阵，每个格子放一件宝贝，每个宝贝贴着价值标签。

地宫的入口在左上角，出口在右下角。

小明被带到地宫的入口，国王要求他只能向右或向下行走。

走过某个格子时，如果那个格子中的宝贝价值比小明手中任意宝贝价值都大，小明就可以拿起它（当然，也可以不拿）。

当小明走到出口时，如果他手中的宝贝恰好是 k 件，则这些宝贝就可以送给小明。

请你帮小明算一算，在给定的局面下，他有多少种不同的行动方案能获得这 k 件宝贝。

输入格式

第一行 3个整数，n,m,k，含义见题目描述。

接下来 n行，每行有 m个整数Ci用来描述宝库矩阵每个格子的宝贝价值。

输出格式
输出一个整数，表示正好取 k个宝贝的行动方案数。

该数字可能很大，输出它对1000000007取模的结果。

数据范围
$1\le n,m\le 50,1\le k\le 12,0\le Ci\le 12$

样例

输入样例1：

2 2 2
1 2
2 1

输出样例1：

2

输入样例2：

2 3 2
1 2 3
2 1 5

输出样例2：

14

题目分析以及算法复杂度分析

题目很容易发现是一道dp的问题，但是本题是比较复杂的。首先二维地图表示我们的dp数组f一定不能少于两维，题目又多了对拿取物品和当前所拿去物品价值的限制，因此我们需要对这两个限制对f数组多加上两维。由于物品的价值是从0开始的，我么为了减少特殊判断，因此将所有的物品数量都加一，因为所求得是方案数，因此并不影响最终的答案。仍然使用yxc的dp分析方法。

步骤是通用的，分成两大部分状态表示和状态计算，然后进一步分析状态表示中的集合和属性、状态计算中的每一个状态

完整代码

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

using namespace std;

const int N = 60, mod = 1000000007;
int f[N][N][13][14], w[N][N]; //手中的宝物件数，最大的宝物价值
int n, m, k;

int main()
{
    cin >> n >> m >> k;
    for(int i = 1; i <= n; i ++)
        for(int j = 1; j <= m; j ++)
        {
            scanf("%d", &w[i][j]);
            w[i][j] ++; //都加一，避免处理边界问题
        }
    
    //最开始只有两种情况，拿或者不拿（第一次一定是可以拿的）
    f[1][1][1][w[1][1]] = 1; //初始化，在（1，1）的位置，拿一件物品，最大价值是w[i][j]。方案赋值为1
    f[1][1][0][0] = 1; //在（1，1）的位置，不物品，最大价值是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++) //拿了几件(这里从0开始是因为可以一件也不拿)
                for(int v = 0; v <= 13; v ++) //当前的最大价值是多少
                {
                    int& t = f[i][j][u][v];
                    t = (t + f[i - 1][j][u][v]) % mod; //这里不建议直接加两个，因为容易爆int，一个一个加即可
                    t = (t + f[i][j - 1][u][v]) % mod;
                    if(u > 0 && v == w[i][j])
                        for(int s = 0; s < w[i][j] ; s ++) 
                        //由于是从小到大枚举的，因此当前的v即是物品的最大价值，因此可以通过每一个比它小的状态得来
                        {
                            t = (t + f[i - 1][j][u - 1][s]) % mod;
                            t = (t + f[i][j - 1][u - 1][s]) % mod;
                        }
                }
        }
            
                
    int res = 0;
    for(int i = 0; i <= 13; i++) res = (res + f[n][m][k][i]) % mod; 
    //因为题目需要的是当手中的物品刚好是k的答案，因此将走完地图并且手中有刚好有k件物品时的方案数相加
   
    cout << res << endl;
    return 0;
}

这种dp问题就是套娃，多加几个状态限制就不太好写了