POJ-1751 Highways

题目链接: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;
}
本站公众号
   欢迎关注本站公众号,获取更多程序园信息
开发小院