//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