1. 程式人生 > >洛谷——P3366 【模板】最小生成樹

洛谷——P3366 【模板】最小生成樹

題目描述

如題,給出一個無向圖,求出最小生成樹,如果該圖不連通,則輸出orz

輸入格式:

第一行包含兩個整數N、M,表示該圖共有N個結點和M條無向邊。(N<=5000,M<=200000)
接下來M行每行包含三個整數Xi、Yi、Zi,表示有一條長度為Zi的無向邊連線結點Xi、Yi

輸出格式:

輸出包含一個數,即最小生成樹的各邊的長度之和;如果該圖不連通則輸出orz

題解:

自己找資料腦補,畢竟只是個模板。

程式碼:

#include<cstdio>
#include <stdlib.h>
#include <algorithm>
using
namespace std; struct arr{ int x,y,z; }a[2000001]; int o,p,n,m,k,tot,ans,f[2000001]; int find(int x){ if (f[x]==x) return f[x]; f[x]=find(f[x]); return f[x]; } bool cmp(arr a,arr b) { return a.z<b.z; } int main(){ scanf("%d%d",&n,&m); for (int i=1;i<=m;i++) scanf
("%d%d%d",&a[i].x,&a[i].y,&a[i].z); sort(a+1,a+1+m,cmp); for (int i=1;i<=n;i++) f[i]=i; tot=1; k=ans=0; while (tot<n){ k++; o=find(a[k].x); p=find(a[k].y); if (o!=p){ ans+=a[k].z; f[o]=p; tot++; } } printf
("%d ",ans); }