1212. 地宫取宝 - 四维dp

1212.地宫取宝 原题链接https://www.acwing.com/problem/content/1214/对于一些基础的二维的动态分析（ f[i][j] ），在题目添加了很多限制条件后会变得很复杂，而这时我们可以考虑增加状态方程的纬度，来简化问题。

本题中并不像其他题让你输出走到右下角后的和最大值，而是要求

• 走到右下角后手中要有k件物品，即限制了在走的过程中对考虑对物品捡与不捡
• 捡物品也不是想捡就捡，只要当手中的所有物品的最大价值C都比当前位置的物品的价值小才可以捡

本题要求的是输出走到右下角时，手中刚好有k件物品的方案数。那么毫无疑问，我们可以给状态方程增加一个维度表示走到位置(i,j)时手中有k件物品，即f[i][j][k]

这时我们就解决了第一个限制条件，对于第二个条件，我们一样想怎样表示当走到(i,j)时手中物品的最大价值呢，我们发现这个最大价值与此时手中的物品数也是有关系的，即捡了k件物品，一定会有一个物品的最大值c，这样第四个维度也就出来了

状态方程：，表示在位置(i,j)时，手中有k件物品的最大值c的方案数，那么最终的结果就是把上c的所有取值求和，因为题目问的是手中有k件物品，而最大价值可以是任何值

代码

#include <bits/stdc++.h>
using namespace std;

const int N = 55;
#define p 1000000007
int a[N][N],f[N][N][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++)
        {
            cin >> a[i][j];
            a[i][j]++;
        }
    
    f[1][1][1][a[1][1]] = 1;
    f[1][1][0][0] = 1;
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
            for(int u=0; u<=K; u++)
                for(int v=0; v<=13; v++)
                {
                    int &val = f[i][j][u][v];
                    //不取(i,j)位置上的宝贝
                    val = (val+f[i-1][j][u][v])%p;
                    val = (val+f[i][j-1][u][v])%p;
                    //取 u>0，且此时宝贝的价值要和w[i][j]一样，我们才计算
                    if( v==a[i][j])
                    {
                        for(int c=0; c<v; c++)
                        {
                            val = (val+f[i-1][j][u-1][c])%p;
                            val = (val+f[i][j-1][u-1][c])%p;
                        }

                    }
                }
    
    int res=0;
    for(int i=1; i<=13; i++)
        res = (res+f[n][m][K][i])%p;

    cout<<res;
    return 0;
}