codeup 建设电力系统 最小生成树问题 采用kruskal方法(选取n-1条边所以此方法适合稀疏图)

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn=501,maxsize=124751;
struct Edge
{
    int u;
    int v;
    int price;

};
vector<struct Edge>ve;
int father[maxn];
int findfather(int x)
{
       if(x==father[x])
              return x;
       else
       {
         int F=findfather(father[x]);//
         father[x]=F;
         return F;

       }
}
void Union(int a,int b)
{
     int fa=findfather(a);
     int fb=findfather(b);
     if(fa!=fb)
              father[fa]=fb;

}
int cmp(struct Edge a,struct Edge b)
{
    return a.price<b.price;

}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
       int n,e,sum=0,cnt=0;
       cin>>n>>e;
       ve.clear();
       //初始化
       for(int i=0;i<n;++i)
              father[i]=i;
       for(int i=0;i<e;++i)
       {
              int a,b,k;
              cin>>a>>b>>k;
              struct Edge edge;
              edge.u=a;
              edge.v=b;
              edge.price=k;
              ve.push_back(edge);
       }
       sort(ve.begin(),ve.end(),cmp);
       //n个顶点要想全部连通 并且不出现还那么至少要有n-1条边
       for(int i=0;i<ve.size();++i)
       {

           int fu=findfather(ve[i].u);
           int fv=findfather(ve[i].v);
           if(fu!=fv)//只要两个端点不在一个联通块里面 就可以加入生成树
           {
              sum+=ve[i].price;
              cnt++;//边数累计 只要达到n-1条就break
              father[fu]=fv;
           }
           if(n-1==cnt)
              break;


       }
       cout<<sum<<endl;



    }
    return 0;
}

http://codeup.cn/problem.php?id=1121