POJ - 1751 Highways （最小生成树）

题目链接：点击打开链接

题意：n个村庄，你需要连接每一个村庄，村庄之间的距离为二维坐标系中点和点的距离。但是已经有了一些路了，问还需要再建那些路来连接村子。

题解：最小生成树，然后先连接已经建好的边，然后再在这个图的基础上再建边。kruskal（）

代码：

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <math.h>
using namespace std;
struct no{
    int x,y;
}node[750+10];
struct ed{
    int u,v;
    double w;
}edge[750*750/2+10],edd[1000];
int cmp(ed a,ed b){
    return a.w<b.w;
}
int n,m,cnt;
int B[750+10];
int fine(int x){
    if(x != B[x])
        B[x] = fine(B[x]);
    return B[x];
}
void bin(int x,int y){
    int xx = fine(x);
    int yy = fine(y);
    if(xx != yy)
        B[xx] = yy;
}
void kru(){
    int i,j;
    for(i = 0;i < m;i ++){
         int xx = fine(edge[i].u);
         int yy = fine(edge[i].v);
         if(xx != yy){
             B[xx] = yy;
             edd[cnt].u = edge[i].u;  // 合并时顺便记录合并了那些变。
             edd[cnt].v = edge[i].v;
             edd[cnt++].w = edge[i].w;
         }
    }
}
int main(){
    scanf("%d",&n);
    for(int i = 1;i <= n;i ++)
        B[i] = i;
    for(int i = 1;i <= n;i ++){
        scanf("%d%d",&node[i].x,&node[i].y);
    }
    m=0;
    for(int i = 1;i <= n;i ++){
        int x1 = node[i].x;
        int y1 = node[i].y;
        for(int j = i + 1 ; j <= n ;j ++){
            int x2 = node[j].x;
            int y2 = node[j].y;
            double d = sqrt(((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)));
            edge[m].u = i;
            edge[m].v = j;
            edge[m ++].w = d;
        }
    }
    cnt = 0;
    sort(edge,edge+m,cmp);
    int k;
    scanf("%d",&k);
    int a,b;
    for(int i = 0 ;i < k;i ++){   // 连接已经建好的路
        scanf("%d%d",&a,&b);
        bin(a,b);
    }
    kru();
    for(int i = 0;i < cnt;i ++)
        printf("%d %d\n",edd[i].u,edd[i].v);
    return 0;
}