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

P3366 【模板】最小生成樹

解釋 truct 技術 題目 bre != union 100% 個數

題目描述

如題,給出一個無向圖,求出最小生成樹,如果該圖不連通,則輸出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

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 using namespace std;
 6 const int MAXN=200001
; 7 struct node 8 { 9 int u,v,w; 10 }edge[MAXN]; 11 int f[MAXN]; 12 int comp(const node & a,const node & b) 13 { 14 return a.w<b.w; 15 } 16 int find(int x) 17 { 18 if(f[x]!=x) 19 f[x]=find(f[x]); 20 return f[x]; 21 } 22 void unionn(int x,int y) 23 {
24 int fx=find(x); 25 int fy=find(y); 26 f[fx]=fy; 27 } 28 int main() 29 { 30 int n,m; 31 scanf("%d%d",&n,&m); 32 for(int i=1;i<=n;i++) 33 f[i]=i; 34 for(int i=1;i<=m;i++) 35 { 36 scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w); 37 } 38 sort(edge+1,edge+m+1,comp); 39 int k=0; 40 int ans=0; 41 for(int i=1;i<=m;i++) 42 { 43 if(find(edge[i].u)!=find(edge[i].v)) 44 { 45 unionn(edge[i].u,edge[i].v); 46 ans+=edge[i].w; 47 k++; 48 } 49 if(k==n-1)break; 50 } 51 if(k!=n-1) 52 printf("orz"); 53 else printf("%d",ans); 54 return 0; 55 }

P3366 【模板】最小生成樹