poj1751 最小生成树

题目链接在这里

题目描述：

有N个乡村，每个乡村的地理位置由二维坐标系表示出来。现已建成M条道路，问还需要在哪些村之间铺路使得村之间相互连通。铺路的成本与村与村之间的距离成正比。

思路分析：

我用的Kruskal方法建立最小生成树，输入M条边时，将这些边连接的两顶点用并查集连接起来。然后暴力算出乡村之间两两的距离，从小到大排序之后扫一遍建边就ok了。

注意的是，M可能是大于N的，当然这并不代表着乡村之间此时就相互连通了，这个时候仍然可能有乡村被孤立。所以不能用n - m来判断还需要建多少个边。

另外也有用优先队列存边的，我超时了，后来改成数组，然后sort，就过了。

代码如下：

#include <cstdio>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
#include <iostream>
#include <algorithm>
using namespace std;
const int MaxN = 760;
const int MaxM = 1010;
struct Point{
    int x, y;
}point[MaxN];

struct Edge{
    int x, y;
    int val;
    bool operator < (const Edge& e) const{
        return val < e.val;
    }
}edge[MaxN * MaxN];
int n, m;
int par[MaxN], r[MaxN];

void init(){
    for(int i = 0; i < MaxN; ++i){
        par[i] = i;
        r[i] = 0;
    }
}

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

void unite(int x, int y){
    x = Find(x);
    y = Find(y);
    if(x == y)  return ;
    if(r[x] < r[y]) par[x] = y;
    else{
        par[y] = x;
        if(r[x] == r[y])    ++r[x];
    }
}

bool check(int x, int y){
    return Find(x) == Find(y);
}

double getDis(int i, int j){
    Point p1 = point[i];
    Point p2 = point[j];
    return (p1.x - p2.x) * (p1.x - p2.x) + (p1.y - p2.y) * (p1.y - p2.y);
}

int main(){
    init();
    scanf("%d", &n);
    for(int i = 1; i <= n; ++i){
        scanf("%d %d", &point[i].x, &point[i].y);
    }
    scanf("%d", &m);
    int x, y;
    for(int i = 0; i < m; ++i){
        scanf("%d %d", &x, &y);
        unite(x, y);
    }

    int cnt = 0;
    for(int i = 1; i <= n; ++i){
        for(int j = 1; j < n; ++j){
            edge[cnt++] = Edge{i, j, getDis(i, j)};
        }
    }
    sort(edge, edge + cnt);
    for(int i = 0; i < cnt; ++i){
        if(check(edge[i].x, edge[i].y)) continue;
        unite(edge[i].x, edge[i].y);
        printf("%d %d\n", edge[i].x, edge[i].y);
    }
    return 0;
}