地宫取宝(多维dp)

地宫取宝

题目链接
X 国王有一个地宫宝库，是 n×m 个格子的矩阵，每个格子放一件宝贝，每个宝贝贴着价值标签。
地宫的入口在左上角，出口在右下角。
小明被带到地宫的入口，国王要求他只能向右或向下行走。
走过某个格子时，如果那个格子中的宝贝价值比小明手中任意宝贝价值都大，小明就可以拿起它（当然，也可以不拿）。
当小明走到出口时，如果他手中的宝贝恰好是 k 件，则这些宝贝就可以送给小明。
请你帮小明算一算，在给定的局面下，他有多少种不同的行动方案能获得这 k 件宝贝。

输入格式

第一行 3 个整数，n,m,k，含义见题目描述。
接下来 n 行，每行有 m 个整数 Ci 用来描述宝库矩阵每个格子的宝贝价值。

输出格式

输出一个整数，表示正好取 k 个宝贝的行动方案数。
该数字可能很大，输出它对 1000000007 取模的结果。

数据范围

1≤n,m≤50,
1≤k≤12,
0≤Ci≤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的.还有人用dfs+记忆化搜索的.我选取的是第一种,看了好几遍别人的代码才搞懂.开四维数组去存储状态.首先是坐标x,y,然后是我们选取的个数,最后是我们选取的最大值,这个最大值一定是最后选取的,按照题意来说.然后我们开始枚举坐标以及当前选取个数,还有拥有的最大值.四重循环,还有内部更新的循环.我们很容易想多状态转移方程,一个点是从上面和左边过来的,那当我们不选当前位的时候,那么就是更新两个的加和.如果我们选的话,一定是满足我们枚举的当前最大值是和当前位是一样的.选完之后去遍历,从小于该最大值的点.更新dp就可以了.最后我们选取的是满足k个物品的各种最大值.最后输出答案.
总结一下, dp[i][j][q][w]代表的是,到达i,j点,拿了q个东西,其中最大值为w的方案个数
核心就是我们的dp状态转移方程:

//不拿当前位
                    dp[i][j][q][w]=(dp[i][j][q][w]+dp[i-1][j][q][w])%mod;
                    dp[i][j][q][w]=(dp[i][j][q][w]+dp[i][j-1][q][w])%mod;
//拿当前位
                    if(q>0&&w==map[i][j])
                    {
                         for(int s=0;s<map[i][j];s++)
                        {
                            dp[i][j][q][w]=(dp[i][j][q][w]+dp[i-1][j][q-1][s])%mod;
                            dp[i][j][q][w]=(dp[i][j][q][w]+dp[i][j-1][q-1][s])%mod;
                        }
                    }

代码实现

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=51;
const int mod=1e9+7;
int dp[maxn][maxn][15][15];
int map[maxn][maxn];
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>>map[i][j];
            map[i][j]++;//因为某个点的值可能为0,处理一下边界情况
        }
    }
    dp[1][1][0][0]=1;
    dp[1][1][1][map[1][1]]=1;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            for(int q=0;q<=k;q++)			//遍历选取的个数
            {
                for(int w=0;w<=13;w++)		//遍历并更新最大值选取的方案数
                {
                    //不拿当前位
                    dp[i][j][q][w]=(dp[i][j][q][w]+dp[i-1][j][q][w])%mod;
                    dp[i][j][q][w]=(dp[i][j][q][w]+dp[i][j-1][q][w])%mod;
                    //拿当前位
                    if(q>0&&w==map[i][j])
                    {
                        for(int s=0;s<map[i][j];s++)	//从比当前位小的最大值更新过来
                        {
                            dp[i][j][q][w]=(dp[i][j][q][w]+dp[i-1][j][q-1][s])%mod;
                            dp[i][j][q][w]=(dp[i][j][q][w]+dp[i][j-1][q-1][s])%mod;
                        }
                    }
                }
            }
        }
    }
    int res=0;
    for(int i=1;i<=13;i++)	//我们加和所有符合选取个数的所有最大值可能
    {
        res=(res+dp[n][m][k][i])%mod;
    }
    cout<<res<<endl;
    return 0;
}

别人的dfs写法,感觉这个很厉害
别人的题解
DFS写法

#include<iostream>
#include<cstdio>
#include<memory.h>
using namespace std;
const int N=51;
const int MOD=1000000007;
int map[N][N]; //记录当前迷宫宝贝的价值
int n,m,k;
int dp[N][N][15][15];//状态记录，比如dp[3][4][5][6]表示对于（3,4）这个点拥有物品数量为5最大价值为6的方案数。

int dfs(int x,int y,int sum,int max){  //搜就完了
      if(dp[x][y][sum][max+1]!=-1) return dp[x][y][sum][max+1]; //记忆化，如果某种状态被记录过则直接拿去用，max+1我解释一下，我么将最初最大价值初始化为-1，但这显然是不正确的，最大价值最低就是0，注意0也是一种状态，我们可以将没有宝贝的格子看成是价值为0的宝贝！！！！

      int t=0;//方案数
      if(x==n-1&&y==m-1){ //走到终点
          if(map[x][y]>max){  //能选最后一个格子中的宝贝
              if(sum==k||sum==k-1) t++; //要么不选，要么选，选的话就要保证自己还能再选一个且只能再选一个
          }
          else if(k==sum) t++;//不能选也是一种方案
          return dp[x][y][sum][max+1]=t; //返回终点的方案数

      }

      if(x+1<n){ //向下走
          if(map[x][y]>max){ //选
              t+=dfs(x+1,y,sum+1,map[x][y]);
              t%=MOD;
          }

          t+=dfs(x+1,y,sum,max); //不选
          t%=MOD;
      }

      if(y+1<m){ //能向右走
          if(map[x][y]>max){ //选
              t+=dfs(x,y+1,sum+1,map[x][y]);
              t%=MOD;
          }

          t+=dfs(x,y+1,sum,max);//不选
          t%=MOD;
      }
      dp[x][y][sum][max+1]=t;//两种情况考虑完后记录当前状态方案数
      return dp[x][y][sum][max+1];

}


int main(){
    cin>>n>>m>>k;
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>map[i][j];
        }
    }

    memset(dp,-1,sizeof(dp));
    dp[0][0][0][0] = dfs(0,0,0,-1);
    cout<<dp[0][0][0][0]<<endl;
    return 0;
}