0
点赞
收藏
分享

微信扫一扫

【luogu3366】 最小生成树模板


(​​http://www.elijahqi.win/2017/07/11/%E3%80%90luogu3366%E3%80%91-%E6%9C%80%E5%B0%8F%E7%94%9F%E6%88%90%E6%A0%91%E6%A8%A1%E6%9D%BF/​​​)
题目描述

如题,给出一个无向图,求出最小生成树,如果该图不连通,则输出orz

输入输出格式

输入格式:

第一行包含两个整数N、M,表示该图共有N个结点和M条无向边。(N<=5000,M<=200000)

接下来M行每行包含三个整数Xi、Yi、Zi,表示有一条长度为Zi的无向边连接结点Xi、Yi

输出格式:

输出包含一个数,即最小生成树的各边的长度之和;如果该图不连通则输出orz

输入输出样例

输入样例#1:

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3
输出样例#1:

7
说明

时空限制:1000ms,128M

数据规模:

对于20%的数据:N<=5,M<=20

对于40%的数据:N<=50,M<=2500

对于70%的数据:N<=500,M<=10000

对于100%的数据:N<=5000,M<=200000

样例解释:

所以最小生成树的总边权为2+2+3=7

#include<cstdio>
#include<algorithm>
#define N 5500
#define M 220000
using namespace std;
int num,n,fa[N];
inline int read(){
int x=0;char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') {
x=x*10+ch-'0';ch=getchar();
}
return x;
}
struct node{
int x,y,w;
}data[M];
inline void insert1(int x,int y,int w){
data[++num].x=x;data[num].y=y;data[num].w=w;
}
inline int find(int x){
return x==fa[x]?x:fa[x]=find(fa[x]);
}
inline bool cmp(node a,node b){
return a.w<b.w;
}
int main(){
freopen("3366.in","r",stdin);
freopen("3366.out","w",stdout);
n=read();num=0;int m=read();
int x,y,z;
for (int i=1;i<=m;++i) x=read(),y=read(),z=read(),insert1(x,y,z);
for (int i=1;i<=n;++i) fa[i]=i;
int tot=0;
sort(data+1,data+1+num,cmp);
//for (int i=1;i<=num;++i )printf("%d %d %d\n",data[i].x,data[i].y,data[i].w);
int i;
long long ans=0;
for (i=1;i<=num;++i){
int p=find(data[i].x),q=find(data[i].y);
if (p!=q) fa[p]=q,++tot,ans+=data[i].w;
// printf("%d %d %d\n",data[i].x,data[i].y,data[i].w);
if (tot==n-1) break;
}
if (tot<n-1) printf("orz");else
printf("%lld",ans);
return 0;
}


举报

相关推荐

0 条评论