先来了解一下什么叫最小生成树
—— 定义:一个有 n 个结点的连通图的生成树是原图的极小连通子图,且包含原图中的所有 n 个结点,并且有保持图连通的最少的边。
—— 通俗的说就是一个能连通所有点且不形成回路的无向图。
再来讲一下求最小生成树的原理
—— 一个图是联通的当且仅当该图上的任意两点相互可达,那我们就可以选取一个点作为所有点的祖先(根),祖先相同的两点必然相互可达。
—— 开始时每个点的祖先都是它自己,已连通的点数为 1 ,每添加一条新边时,判断两端点的祖先是否相同,如不同,则令两点祖先统一,且连通的点数+1,当已连通的点数等于点的总数时,最小生成树已求出,停止加边。
—— 这里还要考虑一个权重问题(也就是总边长要最小),所以在加边前要按权重排个序(从小到大排)
好!话不多说,直接上代码!
#include <bits/stdc++.h>
using namespace std;
const int maxn=200005;
struct node
{
int s,e,l;//储存各边信息,s为边的起点,e为边的终点,l为边的长度
}qaq[maxn];
int fat[maxn];//记录各顶点的祖先
int father(int x)//找各个顶点的祖先,联通的各顶点祖先相同
{
if(fat[x]!=x)
fat[x]=father(fat[x]);
return fat[x];
}
bool cmp(struct node a,struct node b)
{
return a.l<b.l;
}
int main()
{
ios::sync_with_stdio(false);//加快cin与cout的运行速度
int n,m;
cin>>n>>m;//录入顶点数n和边数m
for(int i=1;i<=n;i++)
fat[i]=i;//最开始将各点祖先标记为该点本身
for(int i=1;i<=m;i++)
{
cin>>qaq[i].s>>qaq[i].e>>qaq[i].l;
}
sort(qaq+1,qaq+m,cmp);//将各边信息按长度从小到大进行排序
int sum=1,len=0;//sum用来记录联通的点数,len用来记录最小生成树各边的长度之和
for(int i=1;i<=m;i++)
{
if(sum==n)//最小生成树的边数等于顶点数(即n)-1
break;
if(father(qaq[i].s)!=father(qaq[i].e))//两个联通的点祖先却不同,说明联通的点数可以+1
{
fat[fat[qaq[i].e]]=fat[qaq[i].s];//将两点祖先统一
sum++;
len+=qaq[i].l;
}
}
if(sum<n)//如果联通的点数小于n,说明其不能形成树
cout<<"orz"<<endl;
else
cout<<len<<endl;
return 0;
}