HDU 4424 并查集

点击打开链接

题意：给n个城镇，然后每两个城镇的价值给出，但是u->v的价值是u->v这条路径上的最小值，问从任意一个点出发到其他位置的和的最大值

思路：因为限制条件是路径上的最小值，那么我们按价值排序，从大到小排，就可以避免这个问题了，然后利用并查集来完成，对于两个集合，左集合的价值总和为A,右集合的价值总和为B，现在有一条路径连接这两个集合，那么我们考虑的是将A合并到B还是将B合并到A,我们可以这么比较，连接这两个集合的路径肯定是目前最小的价值，如果A集合元素个数为Ai，那么将A合并到B则B需要加A*Ai，反过来A则加B*Bi，比较哪个大就行了，因为它们一定是合并成一块了，价值大的留下

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=200010;
int f[maxn];
ll num[maxn],r[maxn];
struct edge{
    int u,v;
    ll cost;
}es[maxn];
bool cmp(const edge &a,const edge &b){
    return a.cost>b.cost;
}
int find1(int x){
    if(x!=f[x]) f[x]=find1(f[x]);
    return f[x];
}
ll slove(int n,int k){
    for(int i=0;i<=n;i++) num[i]=0;
    for(int i=0;i<k;i++){
        int u=find1(es[i].u);
        int v=find1(es[i].v);
        if(num[u]+r[v]*(es[i].cost)>=num[v]+r[u]*(es[i].cost)){
            f[v]=u;num[u]+=r[v]*es[i].cost;
            r[u]+=r[v];
        }else{
            f[u]=v;num[v]+=r[u]*es[i].cost;
            r[v]+=r[u];
        }
    }
    return num[find1(1)];
}
int main(){
    int n,u,v;
    ll cost;
    while(scanf("%d",&n)!=-1){
        for(int i=0;i<=n;i++){
            f[i]=i;
            r[i]=1;
        }
        int k=0;
        for(int i=0;i<n-1;i++){
            scanf("%d%d%I64d",&u,&v,&cost);
            es[k].u=u;es[k].v=v;es[k++].cost=cost;
        }
        sort(es,es+k,cmp);
        ll ans=slove(n,k);
        printf("%I64d\n",ans);
    }
    return 0;
}