Conquer a New Region UVA - 1664 (krusal算法变形)

传送门

题意：n个城市形成一棵树，每条边有权值C(i,j)。任意两个点的容量S(i,j)定义为i与j唯一通路上容量的最小值。找一个点(他将成为中心城市)，使得它到所有点的容量之和最大。

题解：前提是树，然后参考克鲁斯卡尔算法，首先初始化并查集，一开始每个集合的代表元素都是中心点，维护每个点集的元素个数cnt[i]以及中心点的容量之和ws[i]，其中这个i是这个元素的代表元，一开始i肯定是所在点集的中心城市，且ws[i]=0。之后把所有边从大到小排序，依次遍历每条边e，使用e将已有两个点集连起来，记这两个点集为A，B，中心点分别是a,b，因为已经排序，此时e一定是来自两个点集的边重权值最小的，而且分别来自A和B的任意两点之间的唯一道路一定经过e，所以通路的容量一定是e的权值。然后考虑A和B合并之后的点集，如果要把b作为中心点，产生的容量是wb=ws[b]+cnt[a]*w，其中w是e的权值。同理a也可以算出来，最后维护出最大值即可。

附上代码：

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;

const int maxn=2e5+50;

ll wss[maxn],cnt[maxn];
int f[maxn];

struct edge{
    int from,to,weight;
};
edge edges[maxn];

bool cmp(edge a,edge b)
{
    return a.weight>b.weight;
}

int find_f(int u)
{
    if(u==f[u]){
        return u;
    }else{
        return f[u]=find_f(f[u]);
    }
}

void mergee(int a,int b,ll v)
{
    f[a]=b;
    wss[b]=v;
    cnt[b]+=cnt[a];
}

int main()
{
    int N;
    while(scanf("%d",&N)==1&&N){
        for(int i=1;i<=N;i++){
            cnt[i]=1;f[i]=i;wss[i]=0;
        }
        for(int i=1;i<N;i++){
            edge &e=edges[i];
            scanf("%d%d%d",&e.from,&e.to,&e.weight);
        }
        sort(edges+1,edges+N,cmp);
        ll ans=0;
        for(int i=1;i<N;i++){
            edge &e=edges[i];
            int a=find_f(e.from),b=find_f(e.to);
            ll wa=wss[a]+cnt[b]*e.weight,wb=wss[b]+cnt[a]*e.weight;
            wb>wa?mergee(a,b,wb):mergee(b,a,wa);
            ans=max(wa,wb);
        }
        printf("%lld\n",ans);
    }
    return 0;
}