hdu 1102 Constructing Roads（ Kruskal ）

http://acm.hdu.edu.cn/showproblem.php?pid=1102

 

思路：把那张图的每个点当做从i到j存到结构体中，然后排序一下，在找出不在集合中的最短边加起来就可以了。

 

代码：

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
int map[105][105];
int p[105];
struct node
{
    int a,b,p;
}a[100005];
bool cmp(node a,node b)
{
    return a.p<b.p;
}
int find(int x)
{
    if(p[x]==x)return p[x];
    else return (p[x]=find(p[x]));
}
int kruskal(int n)
{
    int sum=0;
    for(int i=0;i<n;i++)
    {
        int fx=find(a[i].a);
        int fy=find(a[i].b);
        if(fx!=fy)
        {
            p[fy]=fx;
            sum+=a[i].p;
        }
    }
    return sum;
}
int main()
{
    int m;
    while(scanf("%d",&m)!=EOF&&m)
    {
        int i,j;
        for(i=0;i<=m;i++)
        {
            p[i]=i;
        }
        int k;
        int z=0;
        for(i=1;i<=m;i++)
        {
            for(j=1;j<=m;j++)
            {
                cin>>k;
                if(i>=j)continue;
                a[z].a=i;
                a[z].b=j;
                a[z].p=k;
                z++;
            }
        }
        int t;
        cin>>t;
        for(i=0;i<t;i++)
        {
            int x,y;
            scanf("%d %d",&x,&y);
            int fx=find(x);
            int fy=find(y);
            if(fy!=fx)
                p[fy]=fx;
        }
        sort(a,a+z,cmp);
        int e=kruskal(z);
        cout<<e<<endl;
    }
    return 0;
}