POJ 1751 Highways (kruskal)

题目链接：http://poj.org/problem?id=1751

题意是给你n个点的坐标，然后给你m对点是已经相连的，问你还需要连接哪几对点，使这个图为最小生成树。

这里用kruskal不会超时，用prim应该会超时，特别注意在输入的时候不要多组输入，否则会TLE。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int MAXN = 755;
struct edge {
    int u , v , cost;
}a[MAXN * MAXN];
int x[MAXN] , y[MAXN] , cont , par[MAXN] , cnt , f , ans[MAXN * 2];
bool vis[MAXN][MAXN];

void init(int n) {
    for(int i = 1 ; i <= n ; i++) {
        par[i] = i;
    }
    memset(vis , false , sizeof(vis));
    cont = 0;
    cnt = n;
}

int Find(int n) {
    if(n == par[n])
        return n;
    return (par[n] = Find(par[n]));
}

bool cmp(edge a , edge b) {
    return a.cost < b.cost;
}

int kruskal() {
    for(int i = 1 ; i <= cont ; i++) {
        if(cnt == 1)
            break;
        int u = Find(a[i].u) , v = Find(a[i].v);
        if(u != v) {
            ans[f++] = a[i].u;
            ans[f++] = a[i].v;
            par[u] = v;
            cnt--;
        }
    }
}

int main()
{
    int n , m , u , v;
    scanf("%d" , &n);
    {
        init(n);
        for(int i = 1 ; i <= n ; i++) {
            scanf("%d %d" , x + i , y + i);
        }
        scanf("%d" , &m);
        while(m--) {
            scanf("%d %d" , &u , &v);
            u = Find(u) , v = Find(v);
            vis[u][v] = vis[v][u] = true;
            if(u != v) {
                par[u] = v;
                cnt--;
            }
        }
        for(int i = 1 ; i <= n ; i++) {
            for(int j = 1 ; j < i ; j++) {
                if(vis[i][j])
                    continue;
                cont++;
                a[cont].u = j , a[cont].v = i;
                a[cont].cost = (x[i] - x[j])*(x[i] - x[j]) + (y[i] - y[j])*(y[i] - y[j]);
            }            
        }
        sort(a + 1 , a + cont + 1 , cmp);
        f = 0;
        kruskal();
        for(int i = 0 ; i < f ; i += 2) {
            printf("%d %d
" , ans[i] , ans[i + 1]);
        }
    }
}