HDU 1162 Eddy's picture 最小生成树

题目地址： 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;
}