首页 > 其他 > 详细

普及组模板——最小生成树

时间:2017-10-03 11:54:01      阅读:277      评论:0      收藏:0      [点我收藏+]

题目:【模板】最小生成树(洛谷_3366)

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
inline int read(){
    int t=1,num=0;char c=getchar();
    while(c>9||c<0){if(c==-)t=-1;c=getchar();}
    while(c>=0&&c<=9){num=num*10+c-0;c=getchar();}
    return num*t;
}
const int N=5010,M=200010;
struct edge{int f,t,c;}e[M];
bool cmp(edge a,edge b){return a.c<b.c;}
int fa[N],n,m,ans=0,cnt=1;
void init(){
    for(int i=1;i<=n;i++)fa[i]=i;
}
int find(int x){
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
void unite(int x,int y){
    x=find(x);y=find(y);
    if(x==y)return;
    fa[x]=y;
}
void kru(){
    sort(e+1,e+1+m,cmp);
    for(int i=1;i<=m;i++){
        if(cnt==n)break;
        if(find(e[i].f)==find(e[i].t))continue;
        unite(e[i].f,e[i].t);ans+=e[i].c;cnt++;
    }
}
int main()
{
    n=read();m=read();init();
    for(int i=1;i<=m;i++){
        int x=read(),y=read(),z=read();
        e[i].f=x;e[i].t=y;e[i].c=z;
    }
    kru();
    if(cnt==n)printf("%d\n",ans);
    else puts("orz");
    return 0;
}

本文由Yzyet编写,网址为www.cnblogs.com/Yzyet。非Yzyet同意,禁止转载,侵权者必究。

普及组模板——最小生成树

原文:http://www.cnblogs.com/Yzyet/p/7623448.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!