Hdu 4424 Conquer a New Region 并查集

题意:给出一棵树,树中每条边有一权值,对于从节点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;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值