prim保龄求调,样例可过,悬关!
查看原帖
prim保龄求调,样例可过,悬关!
172136
spencer楼主2024/3/30 10:24
//prim

#include<bits/stdc++.h>
using namespace std;

const int N=5e3+50,M=2e5+50,inf=0x3f3f3f3f;
int a[M],nt[M],tot,h[M],w[M],n,m,vis[M],in[N],de[N],dis[N],ans=0;


void add(int x,int y,int z){
	a[++tot]=y;
	nt[tot]=h[x];
	h[x]=tot;
	w[tot]=z;
	de[x]++,de[y]++;
}


int main(){
	cin>>n>>m;
	for(int i=1,x,y,z;i<=m;i++){
		cin>>x>>y>>z;
		add(x,y,z);
	}
	for(int i=1;i<=n;i++){
		if(de[i]==0){
			cout<<"orz";
			return 0;
		}
	}//判定不连通
	in[1]=1;
	for(int i=h[1];i;i=nt[i]){
		dis[i]=w[i];
	}
	for(int i=1;i<n;i++){
		int mind=inf,u=0;
		for(int j=1;j<=n;j++){
			if(dis[j]!=0&&dis[j]<mind)mind=dis[j],u=j;
		}//选择离集合距离最小的点
		ans+=mind;
		in[u]=1,dis[u]=0;//将u加入集合中
		for(int i=h[u];i;i=nt[i]){
			dis[i]=min(dis[i],w[i]);//更新dis
		}
	}
	cout<<ans;
	
	
	return 0;
}
2024/3/30 10:24
加载中...