最后一个测试点wa了
查看原帖
最后一个测试点wa了
1232696
zbbfans楼主2024/8/18 09:21
#include<bits/stdc++.h>
using namespace std;
int mp[5010][5010];
int vis[5010]={};//标记一个点是否已选 
int d[5010];//d[i]表示已选点集到i点的最短边
void prim(int n){
	for(int i=1;i<=n;i++){
        d[i]=1e9;
    }
	d[1]=0;
	int ans=0;
	//每次挑一个未选点中d值最小的点
	for(int step=1;step<=n;step++){//循环n次,模拟每次确定已选点的过程 
		int mi=INT_MAX,id=-1;
		for(int i=1;i<=n;i++){//枚举所有的未选点,找一个距离最小的点 
			if(vis[i]==0 and d[i]<mi){
				mi=d[i];
				id=i;
			}
		}
		if(id==-1){
			printf("orz\n");
			return ;
		}
		ans+=mi;
		vis[id]=1;
		//枚举所有的未选点 
		for(int i=1;i<=n;i++){
			if(!vis[i] and mp[id][i]<d[i]){
				d[i]=mp[id][i];
			}
		}
	} 
	printf("%d\n",ans);
} 
int main(){
	int n,m,a,b,c;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			mp[i][j]=INT_MAX;
		}
	}
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&a,&b,&c);
		if(c<mp[a][b]){
			mp[a][b]=mp[b][a]=c;
		}
	}
	prim(n);
	return 0;
}
2024/8/18 09:21
加载中...