HDU 1875 畅通工程再续

最小生成树Kruskal

 

原理：

　　将边按权值从小到大排列，从小的开始取，用并查集标记已经到达的点，当取到当前边会形成环则放弃（当前边两个端点都已被标记在同一个连通块内）；否则取当前边，将两个端点所在连通块标记到一起。因为只要判断两个端点是否在同一个连通块内，所以并查集可以路径压缩。

边界：取到n-1条边（成功）或者所有边都已遍历过（失败)。

 

ps：这个间接排序函数看起来挺高大上的~~

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;

const int maxn=105;

int x[maxn],y[maxn];
int v[maxn*maxn],u[maxn*maxn];
double w[maxn*maxn];
int fa[maxn];
int r[maxn*maxn];

bool cmp (int x,int y){
    return w[x]<w[y];
}

int find (int x){
    if (fa[x]==x)
        return x;
    return fa[x]=find (fa[x]);
}

int main (){
    int t;
    scanf ("%d",&t);
    int n;
    while (t--){
        scanf ("%d",&n);
        for (int i=0;i<n;i++){
            scanf ("%d%d",&x[i],&y[i]);
        }
        int tot=0;
        for (int i=0;i<n;i++){
            for (int j=i+1;j<n;j++){
                int s=(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]);
                if (s>=100&&s<=1000000){
                    v[tot]=i;
                    u[tot]=j;
                    w[tot++]=100*sqrt (s*1.0);
                }
            }
            fa[i]=i;
        }
        for (int i=0;i<tot;i++)
            r[i]=i;
        int temp=0;
        sort (r,r+tot,cmp);
        double ans=0;
        for (int i=0;i<tot;i++){
            int fv=find(v[r[i]]);
            int fu=find(u[r[i]]);
            if (fv!=fu){
                ans+=w[r[i]];
                temp++;
                fa[fv]=fu;
            }
        }
        if (temp==n-1)
            printf ("%.1f\n",ans);
        else printf ("oh!\n");
    }
    return 0;
}