prim算法模板题

 【输入】

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

 【输出】

7

【代码】

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
#define fi first
#define se second
const ll inf=0x3f3f3f;
const int N=2e5+10;
vector<pair<int,int>>e[N];
priority_queue<int ,vector<pair<int,int>>,greater<pair<int ,int>>>q;
bool vis[N];
int d[N]; 
int n,m;
int cnt,sum;
void prim(int s)
{
	cnt=0;
	sum=0;
	for(int i=1;i<=n;i++)d[i]=inf;
	d[s]=0;
	q.push({0,s});
	while(!q.empty())
	{
		int tmp=q.top().fi; 
		int u=q.top().se;
		q.pop();
		if(vis[u])continue;
		vis[u]=1;
		cnt++;
		sum+=tmp;
		for(int i=0;i<e[u].size();i++)
		{
			int v=e[u][i].fi;
			int len=e[u][i].se;
			if(d[v]>len)
			{
				d[v]=len;
				q.push({d[v],v});
			}
		}
	}
}
int main()
{
	cin>>n>>m;//表示n个点m条边
	for(int i=1;i<=m;i++)
	{
	   int a,b,c;
	   cin>>a>>b>>c;
	   e[a].push_back({b,c});//先输入点，再输入边权值
	   e[b].push_back({a,c});	
	} 
	prim(1);
	//cout<<cnt<<endl;
 	if(cnt<n)cout<<"orz"<<endl; 
 	else cout<<sum<<endl;
    return 0;	
}