HDU-1863(最小生成树+并查集判断是否连通)

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#define inf 0x3fffffff

int N,M;//N represents the number of points,and M donats the number of edges;
int  visit[105];
int map[105][105];
int dis[105];
int set[105];
int tol;

int find(int x)
{
return set[x]=(set[x]==x?x:find(set[x]));
}

void merge(int a,int b)
{
int x=find(a),y=find(b);
if(x!=y)
{
set[x]=y;
tol--;
}
}

int prim()
{
for(int i=1;i<=N;i++)
{
dis[i]=inf;
}
dis[1]=0;
for(int j=1;j<=N;j++)
{
int t=inf,pos;
for(i=1;i<=N;i++)
{
if(!visit[i]&&t>dis[i])
{
t=dis[i];
pos=i;
}
}
visit[pos]=1;
for(int i=1;i<=N;i++)
{
if(!visit[i]&&dis[i]>map[pos][i]&&map[pos][i]!=0x3f3f3f3f)
{
dis[i]=map[pos][i];
}
}
}
int temp=0;
for(i=1;i<=N;i++)
{
temp+=dis[i];
}
return temp;
}

int main()
{
while(scanf("%d%d",&M,&N),M!=0)
{

int a,b,val;
tol=N-1;
memset(visit,0,sizeof(visit));
memset(map,0x3f,sizeof(map));
for(int j=1;j<=N;j++)
set[j]=j;
for(int i=1;i<=M;i++)
{
scanf("%d%d%d",&a,&b,&val);
merge(a,b);
if(val<map[a][b])
{
map[a][b]=val;
map[b][a]=val;
}
}
if(tol>0)
{
printf("?\n");
continue;
}
int ans=prim();
printf("%d\n",ans);
}
return 0;
}