洛谷 P3366 【模板】最小生成樹
阿新 • • 發佈:2017-11-06
sam blank -s space 包含 ref alt fine pac
P3366 【模板】最小生成樹
題目描述
如題,給出一個無向圖,求出最小生成樹,如果該圖不連通,則輸出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<cstring> #include<iostream> #include<algorithm> #define MAXN 200010 usingnamespace std; int n,m,tot,sum; int fa[MAXN]; struct nond{ int x,y,z; }edge[MAXN]; int cmp(nond a,nond b){ return a.z<b.z; } int find(int x){ if(fa[x]==x) return fa[x]; else return fa[x]=find(fa[x]); } int main(){ scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) fa[i]=i; for(int i=1;i<=m;i++) scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].z); sort(edge+1,edge+1+m,cmp); for(int i=1;i<=m;i++){ int dx=find(edge[i].x); int dy=find(edge[i].y); if(dx==dy) continue; fa[dy]=dx; sum+=edge[i].z; tot++; if(tot==n-1) break; } cout<<sum; }
洛谷 P3366 【模板】最小生成樹