P3366 【模板】最小生成树
题目链接https://www.luogu.com.cn/problem/P3366
1 #include<bits/stdc++.h> 2 using namespace std; 3 const int maxm=200005; 4 int n, m; 5 struct edg{ 6 int x, y, z; 7 friend bool operator < (edg a, edg b){//此处使用友元函数运算符重载，功能同11-13行，但sort中不必使用第三参数见第31行 8 return a.z<b.z; 9 } 10 }; 11 //bool cmp(edg a, edg b){ 12 // return a.z<b.z; 13 //} 14 edg e[maxm]; 15 int fa[5005]; 16 void inif(){ 17 for(int i=1; i<=n; i++) 18 fa[i]=i; 19 } 20 int find_fa(int x){ 21 if(x != fa[x])fa[x]=find_fa(fa[x]); 22 return fa[x];//注意此行刚开始粗心多写了个else导致在洛谷上提交出现RE，但在一本通上提交AC， 费解 23 } 24 int ans; 25 int main() 26 { 27 scanf("%d%d", &n, &m); 28 inif(); 29 for(int i=0; i<m; i++) 30 scanf("%d%d%d", &e[i].x, &e[i].y, &e[i].z); 31 sort(e, e+m); 32 int num=0; 33 for(int i=0; i<m; i++){ 34 int fu=find_fa(e[i].x), fv=find_fa(e[i].y); 35 if(fu != fv) 36 { 37 fa[fu]=fv; 38 ans+=e[i].z; 39 num++; 40 if(num==n-1)break; 41 } 42 } 43 if(num != n-1) 44 printf("orz"); 45 else 46 printf("%d", ans); 47 48 return 0; 49 }
