题目地址: http://acm.hdu.edu.cn/showproblem.php?pid=1162
在一个二维平面坐标上有n个点,每两点之间有一条线连接,求其的最小生成树。
最基础的Kruskal算法,直接上代码。
代码如下:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
using namespace std;
/*
freopen("input.txt","r",stdin);
freopen("output.txt","w",stdout);
*/
int n,m;//n个点,总共有m跳边,单向边
double x[105],y[105];
int parent[105];//父亲节点
struct se
{
int x,y;
double w;
}edge[5055];
bool emp(se a,se b)
{
return a.w<b.w;
}
void UFset()
{
for(int i=0;i<=n;i++)
parent[i]=i;
}
int find(int x)
{
int r=x;
while(x!=parent[x])
x=parent[x];
while(r!=x)//剪枝
{
int j=parent[r];
parent[r]=x;
r=j;
}
return x;
}
void Kruskal()
{
int i,count=0;
double sum=0;
UFset();
for(i=0;i<m;i++)
{
int x=edge[i].x;
int y=edge[i].y;
int fx=find(x),fy=find(y);
if(fx!=fy)
{
count++;
parent[fx]=fy;
sum+=edge[i].w;
if(count==n-1)
break;
}
}
printf("%.2f\n",sum);
}
int main()
{
int i,j;
while(cin>>n)
{
m=0;
for(i=1;i<=n;i++)
{
scanf("%lf%lf",&x[i],&y[i]);
for(j=1;j<i;j++)
{
edge[m].x=i;//第i个点
edge[m].y=j;//第j个点
edge[m].w=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));//i点和j点之间的距离
m++;
}
}
sort(edge,edge+m,emp);//边从小到大排序
Kruskal();
}
return 520;
}