地宫取宝(记忆化搜索)

本文介绍了一道编程题,涉及二维矩阵、动态规划和递归搜索算法。小明在国王的地宫中寻找价值最高的k件宝贝,只能向右或向下移动。通过DFS实现动态规划求解不同取宝方案的数量,并对结果取模输出。题目对输入输出格式、数据范围进行了详细规定,提供了两个测试用例。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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

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

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

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

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

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

输入格式
第一行 3 个整数,n,m,k,含义见题目描述。

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

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

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

数据范围
1≤n,m≤50,
1≤k≤12,
0≤Ci≤12

输入样例12 2 2
1 2
2 1
输出样例12
输入样例22 3 2
1 2 3
2 1 5
输出样例214
#include <iostream>
#include <cstring>
#include <algorithm>
#define x first
#define y second

using namespace std;
typedef long long LL;

const int MOD = 1000000007;
const int N = 60;

int a[N][N];
int n,m,k;
int ans;
int dp[N][N][15][15];

int dfs(int st,int ed,int maxv,int sum){
    
    if(sum>k||st>n||ed>m)return 0;
    
    int& v=dp[st][ed][maxv][sum];
    if(v!=-1)return v;//必须要初始化为-1,因为方案数为0也得返回
    
    
    if(st==n&&ed==m){
        if(sum==k){
            return v=1;
        }
        if(sum==k-1&&maxv<a[n][m]){
            return v=1;
        }
        return v=0;
    }
    int t=0;
    if(a[st][ed]>maxv){
        t=(t+dfs(st+1,ed,a[st][ed],sum+1))%MOD;//拿
        t=(t+dfs(st+1,ed,maxv,sum))%MOD;//不拿
        
        t=(t+dfs(st,ed+1,a[st][ed],sum+1))%MOD;
        t=(t+dfs(st,ed+1,maxv,sum))%MOD;
    }
    else{
        t=(t+dfs(st+1,ed,maxv,sum))%MOD;
        
        t=(t+dfs(st,ed+1,maxv,sum))%MOD;
    }
    return v=t%MOD;
}
int main()
{
    cin >> n >> m >> k;
    
    for (int i = 1; i <= n; i ++ ){
       for (int j = 1; j <= m; j ++ ) {
           cin >> a[i][j];
       }
    }
    
    memset(dp, -1, sizeof dp);
    
    cout<<dfs(1,1,-1,0)<<endl;
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小王子y

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值