POJ1751Highways
题意&思路:
给你n个城市的坐标,m组已经连通的城市,使两两城市连通的需要修的道路的总和最小,输出修建公路的两城市的坐标。
kurskal模板题。
代码:
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<ctype.h>
#include<queue>
#include<set>
#include<stack>
#include<cmath>
#include<map>
#define pii pair<int,int>
#define ll long long
#define cl(x) memset(x,0,sizeof(x))
const int N=1e6+10;
const int mod=1e7+9;
const int maxn=0x3f3f3f3f;
const int minn=0xc0c0c0c0;
const int inf=99999999;
using namespace std;
struct town
{
int x,y;
}a[800];
struct edge
{
int u,v,w;
}b[N];
int fa[800],len=0;
int cmp(edge x,edge y)
{
return x.w<y.w;
}
int find(int x)
{
if(fa[x]!=x)
fa[x]=find(fa[x]);
return fa[x];
}
int judge(int x,int y)
{
int t1=find(x),t2=find(y);
if(t1!=t2)
{
fa[t1]=t2;
return 1;
}
return 0;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,i,j;
cin>>n;
for(i=1;i<=n;i++)
fa[i]=i;
for(i=1;i<=n;i++)
cin>>a[i].x>>a[i].y;
int count=0,m;
cin>>m;
for(i=1;i<=m;i++)
{
int u,v;
cin>>u>>v;
if(judge(u,v))
count++;
}
for(i=1;i<=n;i++)
for(j=i+1;j<=n;j++)
{
int u=find(i),v=find(j);
if(u==v)
continue;
b[++len].u=i;
b[len].v=j;
b[len].w=(a[i].x-a[j].x)*(a[i].x-a[j].x)+(a[i].y-a[j].y)*(a[i].y-a[j].y);
}
sort(b+1,b+len+1,cmp);
for(i=1;i<=len;i++)
{
if(judge(b[i].u,b[i].v))
{
cout<<b[i].u<<" "<<b[i].v<<endl;
count++;
}
if(count==n-1)
break;
}
cout<<endl;
return 0;
}