hdu1223(kruskal)


给的是每两个城市和权值,正好方便用kruskal算法输入。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
struct Node
{
    int s,e,v;
}node[10500];
int father[110],n;
void init()
{
     for(int i=1;i<=n;i++)
        father[i]=i;
}

int find_father(int x)
{
     return x==father[x]? x:find_father(father[x]);
}

int merge_father(int a,int b)
{
     int x=find_father(a);
     int y=find_father(b);
     if(x==y)
        return 0;
     father[y]=x;
     return 1;
}

bool cmp(Node a,Node b)
{
    return a.v<b.v;
}

int main()
{
     while(~scanf("%d",&n)&&n)
     {
         for(int i=0;i<n*(n-1)/2;i++)
             scanf("%d %d %d",&node[i].s,&node[i].e,&node[i].v);
         sort(node,node+n*(n-1)/2,cmp);
         init();
         int sum=0;
         for(int i=0,cnt=0;cnt<n-1;i++)
         {
             if(merge_father(node[i].s,node[i].e))
             {
                 sum+=node[i].v;
                 cnt++;
             }
         }
         printf("%d\n",sum);
     }
}

你可能感兴趣的:(hdu1223(kruskal))