POJ 1751 Highways （kruskal）

题目链接：http://poj.org/problem?id=1751

题目大意：给出n个村庄的坐标 和已经了m条路  现在要求建最短的路  将所有村庄连起来  将这些路输出

思路;这题可以用kruskal来做    通过n个点  枚举出所有的边长建立图 然后kruskal的模板题

#include <iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string.h>
using namespace std;
#define maxn 1010
struct node
{
int x,y;
}s[maxn];//村庄坐标
int u[maxn*maxn/2],v[maxn*maxn/2],w[maxn*maxn/2];
int r[maxn*maxn/2],pre[maxn*maxn/2];
int find(int x)
{
int r=x;
while(pre[r]!=r)
{
r=pre[r];
}
int i=x,j;
while(i!=r)
{
j=pre[i];
pre[i]=r;
i=j;
}
return r;
}
int dis(int x1,int y1,int x2,int y2)
{
return (x2-x1)*(x2-x1)+(y2-y1)*(y2-y1);
}
bool cmp(int x,int y)
{
return w[x]<w[y];
}
void join(int x,int y)
{
int fx=find(x),fy=find(y);
if(fx!=fy)
{
pre[fx]=fy;
}
}
void init()
{
int m;
scanf("%d",&m);
while(m--)
{
int a,b;
scanf("%d%d",&a,&b);
join(a,b);
}
}
bool check(int n)
{
int ans=find(1);
for(int i=2;i<=n;i++)
if(find(i)!=ans)
return false;
return true;
}
void kruskal(int n,int t)
{
for(int i=1;i<=n;i++)
{
pre[i]=i;
}
for(int i=0;i<t;i++)r[i]=i;
sort(r,r+t,cmp);

init();
for(int i=0;i<t;i++)
{
int e=r[i],x=find(u[e]),y=find(v[e]);
if(x!=y)//如果不在一个集合之中  在将其将入并入一个集合  同时输出这条边
{
pre[x]=y;
printf("%d %d\n",u[e],v[e]);
}
if(check(n))//检验是否已经都在一个集合之中
break;
}
}
int main()
{
int a,b;
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&s[i].x,&s[i].y);
}
int t=0;
for(int i=1;i<=n;i++)//任意两个城市之间的距离
{
for(int j=i+1;j<=n;j++)
{
u[t]=i,v[t]=j,w[t]=dis(s[i].x,s[i].y,s[j].x,s[j].y);
t++;
}
}
kruskal(n,t);
return 0;
}