地宫取宝 dfs 记忆化

#include<bits/stdc++.h>
using namespace std;
const int N=55;
const int p=1e9+7;
using ll=long long;
int n,m,k,a[N][N];
int dx[2]={0,1};
int dy[2]={1,0};
int dp[N][N][15][15];

bool pd(int x,int y)
{
    return 1<=x&&x<=n&&1<=y&&y<=m;
}
ll dfs(int x,int y,int mx,int cnt)
{
    if(x==n&&y==m) return (cnt==k);//如果合法就返回值
    if(dp[x][y][mx][cnt]!=-1) return dp[x][y][mx][cnt];//如果已经搜索过就返回值
    int res=0;//每一次搜索的值要将res初始化为0
    for(int i=0;i<2;i++)//进行下和右的移动
    {
        int nx=x+dx[i];
        int ny=y+dy[i];
        if(!pd(nx,ny)) continue;
        if(a[nx][ny]>mx&&cnt<k) res=(res+dfs(nx,ny,a[nx][ny],cnt+1))%p;//如果能拿，这为结果的一种情况（即拿走宝物）
        res=(res+dfs(nx,ny,mx,cnt))%p; //这是不拿的情况
    }
    return dp[x][y][mx][cnt]=res;
}

int main()
{
    memset(dp,-1,sizeof dp);
    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]++;//因为一开始小明手中的宝物为0，mx也为0，如果碰到0的宝物不能拿，这里将所有宝物的价值++，避免发生以上的情况
        }
    }
    cout<<(dfs(1,1,0,0)+dfs(1,1,a[1][1],1))%p;//一开始的初始位置有两种情况，拿或不拿
    return 0;
}