第60期：图论-最小生成树-Prim算法

以luogu的最小生成树模板题为例，学习prim算法

最小生成树浅谈 - 呢没理他 - 博客园

1. P3366 【模板】最小生成树

思路：

朴素prim（链式前向星加边）: 

#include<bits/stdc++.h>
using namespace std;
#define INF INT_MAX 
#define maxn 5003
#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];
//链式前向星加边
void add(int u,int v,int w){
	e[++cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt;
}
//读入数据
void init(){
	cin>>n>>m;
	for(int i=1,u,v,w;i<=m;++i){
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
	}
} 
int prim(){
	//先把dis数组附为极大值
	for(int i=2;i<=n;++i){
		dis[i]=INF;
	} 
	//这里要注意重边，所以要用到min
	for(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
		int minn=INF;//把minn置为极大值
		vis[now]=1;//标记点已经走过
		//枚举每一个没有使用的点
		//找出最小边作为新边
		//注意这里不是枚举now点的所有连边，而是1~n
		for(int i=1;i<=n;i++){
			if(!vis[i]&&minn>dis[i]){
				minn=dis[i];
				now=i;
			}
		} 
		ans+=minn;
		//枚举now的所有连边，更新dis数组
		for(int i=head[now];i;i=e[i].next){
			int v=e[i].v;
			if(dis[v]>e[i].w&&!vis[v]){
				dis[v]=e[i].w;
			}
		} 
	}
	return ans;
}

bool judge(){//根据dis数组判断图是否联通 
	for(int i=1;i<=n;i++){
		if(dis[i]==INF){
			return false;
		}
	}
	return true;
}

int main(){
	init();
	prim();
	if(judge()) cout<<ans<<endl;
	else cout<<"orz"<<endl;
	return 0;
}

堆优化的prim(链式前向星加边)：

#include<bits/stdc++.h>
using namespace std;
#define INF INT_MAX 
#define maxn 5003
#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];
//链式前向星加边
void add(int u,int v,int w){
	e[++cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt;
}

typedef pair<int,int> pii;
priority_queue<pii,vector<pii>,greater<pii>> q; 

//读入数据
void init(){
	memset(dis,127,sizeof dis);
	memset(head,-1,sizeof head);
	cin>>n>>m;
	for(int i=1,u,v,w;i<=m;++i){
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
	}
} 
void prim(){
	dis[1]=0;
	q.push(make_pair(0,1));
	while(q.size()&&tot<n){
		int d=q.top().first,u=q.top().second;
		q.pop();
		if(vis[u]) continue;
		tot++;
		ans+=d;
		vis[u]=1;
		for(int i=head[u];i!=-1;i=e[i].next){
			if(e[i].w<dis[e[i].v]){
				dis[e[i].v]=e[i].w;
				q.push(make_pair(dis[e[i].v],e[i].v));
			}
		}
	}
}

bool judge(){//根据dis数组判断图是否联通 
	for(int i=1;i<=n;i++){
		if(dis[i]==INF){
			return false;
		}
	}
	return true;
}

int main(){
	init();
	prim();
	if(tot==n) cout<<ans<<endl;
	else cout<<"orz"<<endl;
	return 0;
}