DFS学习笔记

题目描述

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

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

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

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

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

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

输入格式

输入一行 3 个整数,用空格分开:n,m,k(1≤n,m≤50,1≤k≤12)。

接下来有 n 行数据,每行有 m 个整数 Ci​(0≤Ci​≤12) 代表这个格子上的宝物的价值。

输出格式

要求输出一个整数,表示正好取 k 个宝贝的行动方案数。该数字可能很大,输出它对 1000000007(109+7) 取模的结果。

输入输出样例

输入 #1复制

复制代码
2 2 2
1 2
2 1

输出 #1复制

复制代码
2

输入 #2复制

复制代码
2 3 2
1 2 3
2 1 5

输出 #2复制

复制代码
14
复制代码
#include<bits/stdc++.h>
#define ll long long
#define endl "\n";
using namespace std; 
int n,m,k;
int mem[55][55][15][15];
int a[55][55];
const ll mod=1e9+7;
ll dfs(int x,int y,int cn,int temp)//行,列,手里的宝藏数,手里的价值最高的宝物
{
    if(x>n||y>m)//行或列超范围
    {
        return 0;
    }
    if(mem[x][y][cn][temp]!=-1)//这条路同情况已经被走过了
    {
        return mem[x][y][cn][temp];//返回上一层
    }
    ll cnt=0;
    if(x==n&&y==m)//走到边界
    {
        if(cn==k||(cn==k-1&&a[x][y]>temp))//宝藏数刚好或宝藏数差一件且满足拿起的要求
        {
            cnt++;cnt%=mod;//情况加一种
        }
    }
    if(a[x][y]>temp)//宝藏比手里的都贵,可以拿起来
    {
        cnt+=dfs(x,y+1,cn+1,a[x][y]);//向右走
        cnt+=dfs(x+1,y,cn+1,a[x][y]);//向下走
    }
    cnt+=dfs(x,y+1,cn,temp);//没有拿起宝藏,向右
    cnt+=dfs(x+1,y,cn,temp);//没有拿起宝藏,向下
    mem[x][y][cn][temp]=cnt%mod;//记录该位置往后的可能,防止重复计算浪费算力
    return mem[x][y][cn][temp];//返回值
    
}
int main()
{
    memset(mem,-1,sizeof(mem));//清空四维数组同时赋初值为"-1"
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            cin>>a[i][j];//读入路径
        }
    }
    cout<<dfs(1,1,0,-1);//从第一行第一列位置开始,手里0个宝藏,手里宝藏价值为-1
    return 0;
}

对室友题解代码的解读,自加注释

相关推荐
Nebula_g8 小时前
C语言应用实例:解方程(二分查找)
c语言·开发语言·学习·算法·二分查找·基础
好望角雾眠8 小时前
第四阶段C#通讯开发-5:TCP
网络·笔记·网络协议·tcp/ip·c#
三品吉他手会点灯8 小时前
stm32f103学习笔记-16-RCC(第2节)-讲解系统时钟配置函数SetSysClockTo72()
笔记·stm32·单片机·嵌入式硬件·学习
仙人掌_lz9 小时前
Kimi Linear 论文阅读笔记:第一次“线性注意力”全面胜过全注意力
论文阅读·笔记
wdfk_prog9 小时前
[Linux]学习笔记系列 -- [kernel]cpu
linux·笔记·学习
li星野11 小时前
打工人日报#20251109
笔记
dxnb2211 小时前
【Datawhale25年11月组队学习:hello-agents+Task1学习笔记】
人工智能·学习
nenchoumi311911 小时前
ROS2 Humble 笔记(四)ROS 的最小工作单元-- Node 节点
笔记·机器人·ros2
weixin_3870021511 小时前
漏洞修复学习之CVE-2024-10976漏洞复现
数据库·sql·学习·安全·postgresql
搞机械的假程序猿12 小时前
普中51单片机学习笔记-流水灯
笔记·学习·51单片机