P12874 [蓝桥杯 2025 国 Python A] 巡逻||题解||图论

题目链接

题解:

首先,我们需要考虑一下整个是一颗树,输入的是起点到终点,还有起点到终点的危险值,算出固定长度下所有路的风险值的总和。

所以直接遍历这棵树,遍历每一个点,1到N,用深搜,一段路长度达到 k 就结束,不够也要结束,为了防止重复,还得开一个布尔数组,防止走回头路,走过的要进行标记已走过,没有走过的要标记未走过。

然后,我们要考虑用什么来存储树的节点,一般存子节点和父节点,我们会采用vector<int>eN,用序号表示父节点,值来表示子节点,这样子不断去遍历

但是,这个题目还需要存储危险值,所以我们可以用结构体直接存储这三个值,或者用向量结构体去存储(具体看代码)

最后,注意数据范围,首先看n的范围到了5000,单个危险值的范围到达10的6次方,危险值的总和范围会超过了Int的范围,所以最后算总和的变量要开范围到long long.

代码如下:

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
int n, k, x, y, z;
long long allsum = 0;
const int N = 1e6;

struct Node {
	int end;
	int wei;
	//终点节点,危险值
};
vector<Node>tree[N];
//开始节点
//不能走回头路,所以需要标记是否走过
bool vis[N];

void dfs(int deep, int sum, int length) {
	if (length == k) {
		//到特地的长度,就可以返回了
		//也可以加
//		sum+=tree[deep].wei;
		allsum += sum;
		return;
	}
	//如果还没到长度
	//遍历该节点的子节点
	for (auto &t : tree[deep]) {
		//如果走过的路就跳过
		if (vis[t.end]) {
			continue;
		}
//		sum += t.wei;
		vis[t.end] = 1;
		dfs(t.end, sum + t.wei, length + 1);
		vis[t.end] = 0;
	}
	return;

}

int main() {

	cin >> n >> k;
	for (int i = 2; i <= n; i++) {
		cin >> x >> y >> z;
		//因为需要反过来经过道路也可以正向经过道路
		//道路没有箭头
		tree[x].push_back({y, z});
		tree[y].push_back({x, z});
		//所以需要双向存储,存储两次开头和终点

	}
	for (int i = 1; i <= n; i++) {
		vis[i] = 1;
		dfs(i, 0, 0);
		vis[i] = 0;
	}
	cout << allsum ;
	return 0;
}

这里是红糖,记录我的小白进化史。

希望能帮到你们,创作不易,如果觉得有帮助可以为我点个赞!!

相关推荐
小欣加油11 小时前
leetcode1926 迷宫中离入口最近的出口
数据结构·c++·算法·leetcode·职场和发展
星恒随风12 小时前
C++ 类和对象入门(五):初始化列表、explicit 和 static 成员详解
开发语言·c++·笔记·学习·状态模式
2601_9618454212 小时前
高考真题试卷电子版|2025高考全科试卷分类下载
考研·面试·蓝桥杯·远程工作·程序员创富·高考
浪客灿心12 小时前
项目篇:模块设计与实现
数据库·c++
牛油果子哥q13 小时前
【C++ STL vector】C++ STL vector 终极精讲:动态数组底层原理、两倍扩容机制、迭代器失效、增删查改、性能剖析与工程避坑指南
开发语言·c++
为何创造硅基生物14 小时前
独占指针的创建std::make_unique 本身自带堆出现
c++
kyle~15 小时前
ROS 2 与 Isaac Sim 联合仿真(一)体系架构、环境选型与基础通信闭环
c++·机器人·nvidia·仿真·ros2
努力努力再努力wz15 小时前
【内存管理与高并发内存池系列】从 mmap 到 malloc:文件映射、匿名映射与 glibc 内存分配机制详解
linux·c语言·数据结构·数据库·c++·qt·链表
八解毒剂15 小时前
数据结构-平衡二叉树——对二叉搜索树的优化
数据结构·c++·算法
起床困难户57516 小时前
条款20:协助完成返回值优化
c++