题目链接:https://vjudge.net/problem/POJ-1751
给出n个点坐标,还有m条已经相连的边,按顺序输出构建最小生成树所需要连的边(没有固定样子,按权值从小到大排好就行)
已经相连的点先用并查集标记,再用Kruskal连接剩下的边
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN=750+10;//点数
const int MAXM=300000+10;//边数
int UF[MAXN];
struct Edge
{
int u,v;
double w;
}edge[MAXM];
int tol; //初始化为0
void addedge(int u,int v,double w)
{
edge[tol].u=u;
edge[tol].v=v;
edge[tol++].w=w;
}
bool cmp(Edge a,Edge b)
{
return a.w<b.w;
}
int find(int x)
{
if(UF[x]==-1) return x;
else return UF[x]=find(UF[x]);
}
void Kruskal(int n)
{
sort(edge,edge+tol,cmp);
int cnt=0;
for(int i=0;i<tol;i++)
{
int u=edge[i].u;
int v=edge[i].v;
double w=edge[i].w;
int t1=find(u);
int t2=find(v);
if(t1!=t2)
{
UF[t1]=t2;
cnt++;
printf("%d %d\n",u,v);
}
if(cnt==n-1) break;
}
}
int x[MAXN],y[MAXN];
int main()
{
int T,n,m,s=0,a,b;
scanf("%d",&n);
tol=0;
for(int i=1;i<=n;i++)
{
scanf("%d%d",&x[i],&y[i]);
for(int j=1;j<i;j++)
addedge(i,j,sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])));
}
memset(UF,-1,sizeof(UF));
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%d%d",&a,&b);
int t1=find(a);
int t2=find(b);
if(t1!=t2)
{
UF[t1]=t2;
s++;
}
}
Kruskal(n-s);
return 0;
}