[每日一题]:Shortest Path(NowCoder)
题目:
题目大意:
给你一颗树,然后将 n (n 确保是偶数) 个点 分成 n / 2 对,使得这 n / 2 对之间的路径长度之和最小。
析题得侃:
Code:
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 2e5 + 10;
typedef long long LL;
// 建反向边一定要开 2 倍空间
int head[maxn],Next[maxn],edge[maxn],ver[maxn];
int size[maxn];
int t,n,tot;
int u,v,w;
LL res = 0;
void add(int u,int v,int w) {
ver[++ tot] = v,edge[tot] = w;
Next[tot] = head[u],head[u] = tot;
return ;
}
void DFS(int s,int fa,LL sum) {
for(int i = head[s]; i; i = Next[i]) {
int y = ver[i];
if(y != fa) {
// 我们传的权值只需要是 该节点到子节点大小即可
DFS(y,s,edge[i]);
// 是当前节点的总大小,所以应该加上 所有子节点的大小
size[s] += size[y];
}
}
// 判断节点大小是否是 奇数
if(size[s] & 1) res += sum;
return ;
}
int main(void) {
scanf("%d",&t);
while(t --) {
// 多组测试,一定要记得初始化
tot = 0;
memset(head,0,sizeof(head));
scanf("%d",&n);
for(int i = 1; i <= n; i ++) {
size[i] = 1;
}
for(int i = 1; i < n; i ++) {
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
add(v,u,w);
}
res = 0;
DFS(1,-1,0);
printf("%lld\n",res);
}
return 0;
}
如果说年轻人未来是一场盛宴的话,那么我首先要有赴宴的资格。