码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2017-10-03 12:02:19      阅读:114      评论:0      收藏:0      [点我收藏+]

标签:最小   space   span   const   ini   line   include   ons   style   

题目:【模板】最小生成树(洛谷_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同意,禁止转载,侵权者必究。

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

标签:最小   space   span   const   ini   line   include   ons   style   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!