洛谷——P3366 【模板】最小生成樹
阿新 • • 發佈:2019-02-08
題目描述
如題,給出一個無向圖,求出最小生成樹,如果該圖不連通,則輸出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);
}