zoj题目链接:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3659
题意:有n个点,n-1条边,边上有权值,现要找出一个点s,使得这个点到所有点的距离和最大,距离的定义是路径上最小的权值。
可以用并查集来搞,把边按权值从大到小排序,因为这条边两边的点不在一个集合,考虑s在两边任一边的情况,确定s点在一边的话,那s到另一个集合的距离和就是边权值乘以另一集合元素个数,然后可以得出合并后这个集合的最大距离和。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
#define BUG cout<<"BUG"<<endl;
#define clr(arr,v) memset(arr,v,sizeof(arr))
typedef long long LL;
const int M = 200005;
struct Edge{
int a,b;
LL c;
};
bool cmp(const Edge a,const Edge b)
{
return a.c > b.c;
}
int father[M];
LL sum[M],cnt[M];
Edge ed[M];
int find(int x){
if(father[x] == -1) return x;
return father[x] = find(father[x]);
}
void Union(int a,int b){
father[b] = a;
}
int main()
{
//freopen("input.txt","r",stdin);
int n;
while(cin>>n)
{
int a,b,c;
fill(cnt,cnt+M,1);
clr(father,-1);
clr(sum,0);
for(int i = 0;i < n-1;++i)
{
cin>>ed[i].a>>ed[i].b>>ed[i].c;
}
sort(ed,ed+n-1,cmp);
for(int i = 0;i < n-1;++i)
{
int root_a = find(ed[i].a);
int root_b = find(ed[i].b);
if(root_a != root_b)
{
if(sum[root_a]+cnt[root_b]*ed[i].c > sum[root_b]+cnt[root_a]*ed[i].c)
{
sum[root_a] += cnt[root_b] * ed[i].c;
cnt[root_a] += cnt[root_b];
Union(root_a,root_b);
}
else
{
sum[root_b] += cnt[root_a] * ed[i].c;
cnt[root_b] += cnt[root_a];
Union(root_b,root_a);
}
}
}
int root = find(1);
cout<<sum[root]<<endl;
}
return 0;
}
参考代码: