Prim求助

#include<bits/stdc++.h>
using namespace std;
int cost[5005][5005];
int dis[5005];
int n,m;
bool vis[5005];
int ans=0;
void prim(){
	for(int i=1;i<=n;i++){
		dis[i]=cost[i][1];
	}
	for(int i=1;i<n;i++){
		int minp=2147483647,mj;
		for(int j=2;j<=n;j++){
			if(minp>dis[j]&&!vis[j]){
				minp=dis[j];
				mj=j;
			}
		}
		if(minp==2147483647){
			cout<<"orz";
			exit(0);
		}
		ans+=minp;
		vis[mj]=true;
		for(int j=2;j<=n;j++){
			if(!vis[j]&&dis[j]>dis[mj]+cost[mj][j]){
				dis[j]=dis[mj]+cost[mj][j];
			}
		}
	}
}
int main(){
	scanf("%d%d",&n,&m);
	memset(cost,63,sizeof(cost));
	memset(dis,63,sizeof(dis));
	for(int i=1;i<=m;i++){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		cost[x][y]=cost[y][x]=z;
	}
	for(int i=1;i<=n;i++){
		cost[i][i]=0;
	}
	prim();
	cout<<ans;
	return 0;
}