UVA 10397 - Connect the Campus

这是求最小生成树的一道题，搬了白书上的Kruskal算法，再一次见到了间接排序。

值得注意的是得先用并查集将给的已经有连接的点合并。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#define MAXD 760

double x[MAXD], y[MAXD], w[10000 * MAXD], ans;
int p[MAXD], u[MAXD*MAXD], v[MAXD*MAXD], r[MAXD*MAXD];
int N, M, tt, nx, ny, a, b;
double dist( int i, int j)
{
    return sqrt( (x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j]));
}

int find_set( int x)
{
    return p[x] == x ? x : ( p[x] = find_set(p[x]) );
}

// 间接排序函数
int cmp( const void *_p, const void *_q)
{
    int *p = (int *)_p;
    int *q = (int *)_q;
    return w[*p] > w[*q] ? 1 : - 1;
}

void init()
{
    for( int i = 0; i < N; i ++)
        scanf( "%lf%lf", &x[i], &y[i]);
    tt = 0;
    for( int i = 0; i < N; i ++)
        for( int j = i + 1; j < N; j ++)
        {
            u[tt] = i;
            v[tt] = j;
            w[tt ++] = dist( i, j);
        }
}

void Kruskal()
{
    ans = 0.0;
    // 初始化并查集和边序号
    for( int i = 0; i < N; i ++)
        p[i] = i;
    for( int i = 0; i < tt; i ++)
        r[i] = i;
    qsort( r, tt, sizeof(r[0]), cmp);
    scanf( "%d", &M);
    while( M --)
    {
        scanf( "%d%d", &a, &b);
        a --; b --;
        nx = find_set( a);
        ny = find_set( b);
        if( nx != ny)
            p[nx] = ny;
    }
    for( int i = 0; i < tt; i ++)
    {
        int e = r[i];
        nx = find_set( u[e]);
        ny = find_set( v[e]);
        if( nx != ny) {
            ans += w[e];
            p[nx] = ny;
        }
    }
}

int main()
{
    while( scanf( "%d", &N) == 1)
    {
        init();
        Kruskal();
        printf( "%.2f\n", ans);
    }
    return 0;
}