文章目录
- 关于时间复杂度:
- Kruskal讲解
- Kruskal模板
- Prim讲解
- Prim模板
昨天做蓝桥杯的题,最后一题最小生成树,但好久没用一下子生疏了。。。
又过了几天离散学到了最小生成树。。。趁此复习一波
最小生成树两种方法:
Prim和Kruskal
关于时间复杂度:
prim:该算法的时间复杂度为O(n2)。与图中边数无关,该算法适合于稠密图。
kruskal:需要对图的边进行访问,所以克鲁斯卡尔算法的时间复杂度只和边又关系,可以证明其时间复杂度为O(eloge)。适合稀疏图。
Kruskal讲解
利用并查集来求
连通块:无向图中相互连通 的一些点
具体步骤:将一个连通块当做一个集合,然后按照边的大小(从小到大)进行排序,一开始认为每个点都是孤立的,自身是一个集合。
然后按照顺序枚举每一个边,如果这个边连接了不同的集合,就用并查集将两个集合合并在一起,就成一个集合了。并将这条边加入最小生成树。如果这条边连接的是一个集合,就直接跳过,直到n-1条边为止
Kruskal模板
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
using namespace std;
int n,m,i,j,u,v,total;
struct edge{int start,to;long long val;
}bian[2000005];
int f[100000];
long long ans;int find(int x)//并查集部分
{if (f[x]==x) return x; else return f[x]=find(f[x]);}bool cmp(edge a,edge b)//结构体快排时用到的
{return a.val<b.val;
}inline void kruskal()//最小生成树
{for(int i=1;i<=m;i++){u=find(bian[i].start);v=find(bian[i].to);if(u==v) continue;//判断在不在同一个并查集里面,在就下一个循环ans+=bian[i].val;//不在,就加上f[u]=v;//连接两个并查集total++;if(total==n-1) break;//当形成了最小生成树后,退出(之后做的也没用了)}
}
int main()
{scanf("%d%d",&n,&m);for(i=1;i<=n;i++) f[i]=i;for(i=1;i<=m;i++){scanf("%d%d%d",&bian[i].start,&bian[i].to,&bian[i].val);}sort(bian+1,bian+m+1,cmp);//快排边长kruskal();printf("%d",ans);return 0;
}
Prim讲解
Prim中白点代表已经进入最小生成树的点,蓝点代表未进入最小生成树的二点
MST表示最小生成树权值
min[v]表示蓝点v与白点相连的最小边权值
Prim算法每次回循环将一个蓝点u变成白点,并且此蓝点与白点相连的最小边权值min[u]还是当前所有蓝点中最小的,这样相当于向生成树中添加了n-1次最小的边,最后的到的肯定是最小生成树
步骤:
(a)初始化:min=inf min[1]=0 MST=0
(b)for枚举点
1.寻找min[u]最小的蓝点u
2.将u标记为白点,说明已经进入最小生成树
3.MST+这条边min[u]
4.for枚举与u相连的蓝点v
if(w[u][v]<min[v])min[v]=w[u][v]
不断更新边权
5.输出答案
Prim模板
#include<bits/stdc++.h>
using namespace std;
#define re register
#define il inline
il int read()
{re int x=0,f=1;char c=getchar();while(c<'0'||c>'9'){if(c=='-') f=-1;c=getchar();}while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();return x*f;
}//快读,不理解的同学用cin代替即可
#define inf 123456789
#define maxn 5005
#define maxm 200005
struct edge
{int v,w,next;
}e[maxm<<1];
//注意是无向图,开两倍数组
int head[maxn],dis[maxn],cnt,n,m,tot,now=1,ans;
//已经加入最小生成树的的点到没有加入的点的最短距离,比如说1和2号节点已经加入了最小生成树,那么dis[3]就等于min(1->3,2->3)
bool vis[maxn];
//链式前向星加边
il void add(int u,int v,int w)
{e[++cnt].v=v;e[cnt].w=w;e[cnt].next=head[u];head[u]=cnt;
}
//读入数据
il void init()
{n=read(),m=read();for(re int i=1,u,v,w;i<=m;++i){u=read(),v=read(),w=read();add(u,v,w),add(v,u,w);}
}
il int prim()
{//先把dis数组附为极大值for(re int i=2;i<=n;++i){dis[i]=inf;}//这里要注意重边,所以要用到minfor(re int i=head[1];i;i=e[i].next){dis[e[i].v]=min(dis[e[i].v],e[i].w);}while(++tot<n)//最小生成树边数等于点数-1{re int minn=inf;//把minn置为极大值vis[now]=1;//标记点已经走过//枚举每一个没有使用的点//找出最小值作为新边//注意这里不是枚举now点的所有连边,而是1~nfor(re int i=1;i<=n;++i){if(!vis[i]&&minn>dis[i]){minn=dis[i];now=i;}}ans+=minn;//枚举now的所有连边,更新dis数组for(re int i=head[now];i;i=e[i].next){re int v=e[i].v;if(dis[v]>e[i].w&&!vis[v]){dis[v]=e[i].w;}}}return ans;
}
int main()
{init();printf("%d",prim());return 0;
}