HDU1233最小生成树之kruskal算法加并查集

最小生成树之kruskal算法加并查集

#include<stdio.h>
#include<string.h>
int a[110][110];
int b[110];
int main()
{
int N,i,j,x,y,len,min,mx,my,l,temp,num;
while(scanf("%d",&N)&&N)
{
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
l=0;
num=0;
for(i=0;i<N*(N-1)/2;i++)
{
scanf("%d%d%d",&x,&y,&len);
a[x][y]=len;
a[y][x]=len;
a[x][x]=0;
a[y][y]=0;
}
while(l<N-1)
{
min=-1;
for(i=1;i<=N;i++)
{
for(j=1;j<=i;j++)
{
if(min==-1&&a[i][j]!=0)
{
min=a[i][j];
mx=i;
my=j;
}
if(a[i][j]<min&&a[i][j]!=0)
{
min=a[i][j];
mx=i;
my=j;
}
}
}
if(b[mx]==0&&b[my]==0)
{
l++;
a[mx][my]=0;
b[mx]=l;
b[my]=l;
num+=min;
}
else if(b[mx]==0&&b[my]!=0)
{
l++;
a[mx][my]=0;
b[mx]=b[my];
num+=min;
}
else if(b[mx]!=0&&b[my]==0)
{
l++;
a[mx][my]=0;
b[my]=b[mx];
num+=min;
}
else if(b[mx]!=0&&b[my]!=0&&b[mx]!=b[my])
{
l++;
a[mx][my]=0;
temp=b[my];
for(j=1;j<=N;j++)
{
if(b[j]==temp) b[j]=b[mx];
}
num+=min;
}
else if(b[mx]!=0&&b[my]!=0&&b[mx]==b[my])
{
a[mx][my]=0;
}
}
printf("%d\n",num);
}
return 0;
}