P3366 【模板】最小生成树(洛谷）

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。

输入格式

第一行包含两个整数 N,M，表示该图共有 N 个结点和 M 条无向边。

接下来 M 行每行包含三个整数 Xi​,Yi​,Zi​，表示有一条长度为 Zi​ 的无向边连接结点 

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define PII pair<int,int >;
#define int long long 
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int N = 1e6+5;
int acc[N];
pair<int,pair<int,int>>va[N];//va[i]zhi di i tiao bian
int n,m;
int find(int x)
{if(acc[x]==x) return x;else return acc[x]=find(acc[x]);
}
void krukal()
{int cnt=0,sum=0;//sum   bian quan zong he,cnt bian de ge shu;for(int i=1;i<=n;i++) acc[i]=i;for(int i=1;i<=m;i++){int a=va[i].se.fi,b=va[i].se.se,c=va[i].fi;int t1=find(a),t2=find(b);if(t1!=t2){acc[t1]=t2;sum+=c;cnt++;}}if(cnt==n-1) cout<<sum;else cout<<"orz";
}
signed main()
{IOS;cin>>n>>m;for(int i=1;i<=m;i++){int a,b,c;cin>>a>>b>>c;va[i]={c,{a,b}};}sort(va+1,va+1+m);krukal();return 0;
}

Xi​,Yi​。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

输入输出样例

输入 #1复制

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出 #1复制

7

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define PII pair<int,int >;
#define int long long 
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int N = 1e6+5;
int acc[N];
pair<int,pair<int,int>>va[N];//va[i]zhi di i tiao bian
int n,m;
int find(int x)
{if(acc[x]==x) return x;else return acc[x]=find(acc[x]);
}
void krukal()
{int cnt=0,sum=0;//sum   bian quan zong he,cnt bian de ge shu;for(int i=1;i<=n;i++) acc[i]=i;for(int i=1;i<=m;i++){int a=va[i].se.fi,b=va[i].se.se,c=va[i].fi;int t1=find(a),t2=find(b);if(t1!=t2){acc[t1]=t2;sum+=c;cnt++;}}if(cnt==n-1) cout<<sum;else cout<<"orz";
}
signed main()
{IOS;cin>>n>>m;for(int i=1;i<=m;i++){int a,b,c;cin>>a>>b>>c;va[i]={c,{a,b}};}sort(va+1,va+1+m);krukal();return 0;
}