题目链接:点击打开链接
题意: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;
}