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

题目链接

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

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

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

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

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

代码如下：

#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;
}

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

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