
题解:
首先,我们需要考虑一下整个是一颗树,输入的是起点到终点,还有起点到终点的危险值,算出固定长度下所有路的风险值的总和。
所以直接遍历这棵树,遍历每一个点,1到N,用深搜,一段路长度达到 k 就结束,不够也要结束,为了防止重复,还得开一个布尔数组,防止走回头路,走过的要进行标记已走过,没有走过的要标记未走过。
然后,我们要考虑用什么来存储树的节点,一般存子节点和父节点,我们会采用vector<int>e[N],用序号表示父节点,值来表示子节点,这样子不断去遍历
但是,这个题目还需要存储危险值,所以我们可以用结构体直接存储这三个值,或者用向量结构体去存储(具体看代码)
最后,注意数据范围,首先看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;
}
这里是红糖,记录我的小白进化史。
希望能帮到你们,创作不易,如果觉得有帮助可以为我点个赞!!