Kruskal 算法 从最小边开始,将两个顶点加入同一个集合。
第二小的边,看两个顶点是否已经是连通的了。如果不是连通的,将这个边加入最小生成树中,合并两个顶点。
关于查询是否连通,呃,并查集优化find(
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+9;
struct node {
int u,v,w;//输入三个整数 u,v,w表示u村到v村需要w米距离
}t[N];
bool cmp(node a,node b){//自定义 边从小到大排序
return a.w<b.w;
}
int fa[N];
//查询祖宗
int find(int x) {
if( fa[x]==x ) return fa[x];
else{
fa[x]=find(fa[x]);
return fa[x];
}
}
//合并
void merge(int a,int b){
int a_fa=find(a),b_fa=find(b);
if( a_fa != b_fa )
fa[a_fa]=b_fa;
}
int main()
{
int n,m; cin>>n>>m;
for(int i=0;i<n;i++) fa[i]=i;//初始化
for(int i=1 ;i<=m;i++) {
int u,v,w; cin>>u>>v>>w;//输入三个整数 u,v,w表示u村到v村需要w米距离
t[i]={u,v,w};
}
sort(t+1,t+1+m,cmp);//把m条路的距离 从小到大排序,快排,mlogm
int sum=0,cn=0;
for(int i=1;i<=m;i++) {
int u=t[i].u, v=t[i].v ,w=t[i].w;//每次选择最小的边
if( find(u) != find(v) ) {//如果这条边的两个顶点 不是相同的祖宗 = 两个顶点 目前尚未没有连接
cn++;
merge(u,v); //合并集合,代表能连通了
sum=sum+w; //将选取的边加入最小生成树中
}
}
if(cn<n-1) //特殊情况,原图本不是连通的
cout<<"NO\n";
else
cout<<sum<<endl;
return 0;
}
Prim 算法 从任意取一个顶点,找它的边中最小的,连接下一个点,把两个点看成一个整体。
选整体周边中最小的 边,三个顶点融为一个整体 。
再往外扩散,找新的4个顶点整体周边最小的边,5个顶点连为一体。
看成一个整体 对外寻找,周边