prim模板题 最小生成树

//prim算法 
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f
const int N=1e4+10;
int cnt=0,n,m,ans=0,h;
struct edge{
	int v,w;
};
vector<edge>e[N];//e[N]是来读入题目所给的图 
int d[N],vis[N];//d[N]表示以最近更新的节点为起点跟对应 终点的边权值（距离） 
//vis[N]记录点是否被记录，记录后的点就是最新更新的节点 
bool prim(int s)//s表示起点 
{
	for(int i=0;i<=n;i++)d[i]=inf;
  	d[s]=0;	
    for(int i=0;i<=n;i++)
    {
     int u=0;
	 for(int j=1;j<=n;j++)
	   if(!vis[j]&&d[j]<d[u])u=j;
	 vis[u]=1;
	 ans+=d[u];//这条边一定是最短距离 
	 if(d[u]!=inf) cnt++;//确定这个最小生成树是联通的，联通的话说明cnt=n;	
	 for(auto ed:e[u])//以新的节点为参照点，再更新各个点距离整体更近的点 
	 {
	 	int v=ed.v,w=ed.w;
	 	if(d[v]>w)d[v]=w;
	 }
    }	
  return cnt;
} 
int main()
{
	int a,b,c;
	cin>>n>>m;
	for(int i=0;i<m;i++)
	{
		cin>>a>>b>>c;
		if(i==0) h=a;
		e[a].push_back({b,c});//双向建边 
		e[b].push_back({a,c}); 
	}
	prim(h);
	int k=inf;
	if(cnt==n)cout<<ans-k<<endl; 
	else cout<<"orz"<<endl;
    return 0;	
} 

 【输入】

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3