图_最小生成树_Prim

#include<bits/stdc++.h>

using namespace std;

const int NN=500005;

int N,M,n,s,ans,head[NN],flg[NN];

struct node{
	int x,d,nex;
}a[NN];

void add(int x,int y,int z){
	n++;
	a[n].x=y;
	a[n].d=z;
	a[n].nex=head[x];
	head[x]=n;
}

struct cmp{
	bool operator()(node a,node b) { 
		return  a.d>b.d; 
	}
};

priority_queue<node,vector<node>,cmp>q;

int main() {
	//freopen("x.txt","r",stdin);
	cin>>N>>M;
	for(int i=1;i<=M;++i){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z),add(y,x,z);//无向图 
	}
	for(int i=head[1];i;i=a[i].nex){//随便从哪个点开始，我这里直接从1号节点开始，将与之链接的边全部加入堆中 
		q.push(a[i]);
	}
	flg[1]=1,s=1;
	while(!q.empty()){
		node t=q.top();//小根堆，每次取已纳入最小生成树中点向外衍生边中最短的一条 
		q.pop();
		int x=t.x,d=t.d;
		if(flg[x]) continue;//已经在最小生成树中则跳过 
		ans+=d;//生成树总长 
		++s;//s表示已纳入多少个点 
		flg[x]=1;//标记已纳入最小生成树 
		for(int i=head[x];i;i=a[i].nex){//将这个点向外连的边都加入堆中候选 
			if(flg[a[i].x]) continue;
			q.push(a[i]);
		}
	}
	if(s==N)//这时图是联通的 
		cout<<ans;
	else//图不连通 
		cout<<"orz";
}