poj 1679 The Unique MST （最小生成树）

#include"stdio.h"
#include"string.h"
#define INF 999999999
int map[101][101],mark[101],f[101],n;
void prime()
{
int i,j,k,min,flag,cnt,ans;
memset(mark,0,sizeof(mark));
for(i=1;i<=n;i++)
f[i]=map[1][i];
f[1]=0;mark[1]=1;
ans=0;flag=0;
for(i=1;i<n;i++)
{
min=INF;
for(j=1;j<=n;j++)
{
if(!mark[j]&&f[j]<min)
{
min=f[j];k=j;
}
}
cnt=0;
for(j=1;j<=n;j++)
{
if(mark[j]&&min==map[k][j])
cnt++;
}
if(cnt>1)
{
flag=1;break;
}
ans+=min;
mark[k]=1;
for(j=1;j<=n;j++)
{
if(!mark[j]&&map[k][j]<f[j])
f[j]=map[k][j];
}
}
if(flag)
printf("Not Unique!\n");
else
printf("%d\n",ans);
}
int main()
{
int t,i,j,a,b,c,m;
scanf("%d",&t);
while(t--)
{
memset(map,76,sizeof(map));
scanf("%d%d",&n,&m);
for(i=0;i<m;i++)
{
scanf("%d%d%d",&a,&b,&c);
if(map[a][b]>c)
map[a][b]=map[b][a]=c;
}
prime();
}
return 0;
}