ccf 201412-4 最优灌溉 最小生成树裸题

点击打开链接

//kruskal+并查集
#include <iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxv=1000+5;
const int maxe=1e5+5;
struct Edge{
int from,to,weight;
};
bool operator<(const Edge E1,const Edge E2)
{
    return E1.weight<E2.weight;
}
int n,m;
int parent[maxv];
Edge edges[maxe];
int find(int x)
{
    return x==parent[x]?x:parent[x]=find(parent[x]);
}
void init()
{
    for(int i=1;i<=n;i++)
        parent[i]=i;
}
int main()
{
    scanf("%d%d",&n,&m);
    init();
    int cnt=0;
    for(int i=0;i<m;i++)
    {
        int v1,v2,dist;
        scanf("%d%d%d",&v1,&v2,&dist);
        edges[cnt].from=v1;
        edges[cnt].to=v2;
        edges[cnt].weight=dist;
        cnt++;
    }
    sort(edges,edges+cnt);
    LL sum=0;
    for(int i=0;i<cnt;i++)
    {
        int p1=find(edges[i].from);
        int p2=find(edges[i].to);
        if(p1==p2) continue;
        parent[p1]=p2;
        sum+=edges[i].weight;
    }
    cout<<sum<<endl;
    return 0;
}