使用Prim算法

使用邻接表存储

代码如下：

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int N=5005,M=400010,INF=0x3f3f3f3f;
int dist[N];//某元素到树的距离 
int e[M],ne[M],h[N],w[M],idx;
bool st[N];//状态表示，是否使用过 
int n,m;

void add(int a,int b,int va)//邻接表存储 
{
	e[idx]=b;//存储元素 
	ne[idx]=h[a];//存储下一个元素的下标 
	w[idx]=va;//存储距离 
	h[a]=idx++; 
}

int prim(void)
{
	memset(dist,0x3f,sizeof dist);//全部初始化为无穷 
	int res=0;//答案 
	for(int i=0;i<n;i++)//找够n个点元素 
	{
		int t=-1;
		for(int j=1;j<=n;j++)
		{
			if(!st[j]&&(t==-1||dist[j]<dist[t]))
			{
				t=j;;
			}
		}
		st[t]=true;//标记为找过 
		if(i)res+=dist[t];//加上距离 
		if(i&&dist[t]==INF)return -123456789;//如果距离为无穷，说明不连通 
		for(int k=h[t];k!=-1;k=ne[k])//更新距离 
		{
			int p=e[k];
			dist[p]=min(w[k],dist[p]);
		}
		dist[t]=0; //******一步神奇的代码，删除后16分，加上AC 
	}
	return res;
}
int main()
{
	cin>>n>>m;
	memset(h,-1,sizeof h);//初始链表头为-1 
	for(int i=1;i<=m;i++)//读入 
	{
		int a,b,va;
		cin>>a>>b>>va;
		add(a,b,va);//双向 
		add(b,a,va);
	}
	int t=prim();
	if(t==-123456789)
	{
		cout<<"orz"<<endl;
	}else
	{
		cout<<t<<endl;
	}
	return 0;
} 

为什么那条将自己的距离更新为0会影响答案呢？ 而且根据评测状况，除了16分的一个测试点，其余全部输出了orz，也就是符合了那条距离无穷的判断条件

求解