题意:给出一棵树,树中每条边有一权值,对于从节点i 到节点 j 的路径,他的权值为该路径上最小的边的权值,问确定一个中心节点,使得该中心点到其他点的权值之和最大
思路:由于权值只和路径上边的最小值有关,所以我们把边的权值从大到小排序,每次并查集合并时取最大值即可
代码:
#include <iostream>
#include <stdio.h>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
const int M=200005;
const int mod=1e9+7;
int fa[M];
long long sum[M];
int num[M];
struct node
{
int a,b;
long long w;
} edge[M];
int find(int x)
{
if(x==fa[x])return x;
fa[x]=find(fa[x]);
return fa[x];
}
bool cmp(node a,node b)
{
return a.w > b.w;
}
void Union(int a,int b,long long v)
{
fa[a]=b;
num[b]+=num[a];
sum[b]+=v;
}
int main()
{
int n,a,b,c;
while(scanf("%d",&n)!=EOF)
{
for(int i=1; i<=n; i++)
{
fa[i]=i;
num[i]=1;
sum[i]=0;
}
for(int j=1; j<n; j++)
{
scanf("%d%d%I64d",&edge[j].a,&edge[j].b,&edge[j].w);
}
sort(edge+1,edge+n,cmp);
for(int j=1; j<n; j++)
{
int aa=find(edge[j].a);
int bb=find(edge[j].b);
if(aa==bb)continue;
long long s1=sum[aa]+num[bb]*edge[j].w;
long long s2=sum[bb]+num[aa]*edge[j].w;
if(s1>s2)
{
Union(bb,aa,s1-sum[aa]);
}
else
{
Union(aa,bb,s2-sum[bb]);
}
}
printf("%I64d\n",sum[find(1)]);
}
return 0;
}