prim板子

求最小生成树,(指的是在一个加权无向图中找到一个边的子集,使得这个子集构成的树包含图中的所有顶点,并且边的权重之和最小。)
用于稠密图,复杂度O(mlogm)

int n,m,cnt,ans;
vector<pii>e[maxn];//{点,权重}
int d[maxn],vis[maxn];
priority_queue<pii>q;//{-距离,点},便于找距离最近的点 
bool prim(int s){
	for(int i=0;i<=n;i++)d[i]=inf;
	d[s]=0;q.push({0,s});
	while(!q.empty()){
		int u=q.top().se;q.pop();
		if(vis[u])continue;
		vis[u]=1;
		ans+=d[u];cnt++;
		for(auto ed:e[u]){
			int v=ed.fi,w=ed.se;
			if(d[v]>w){
				d[v]=w;
				q.push({-d[v],v});
			}
		}
	}
	return cnt==n//是否连通;
}
posted @ 2025-02-27 20:47  Marinaco  阅读(24)  评论(0)    收藏  举报
//雪花飘落效果