poj 1679 prim

判断最小生成树唯不唯一

根据prim算法，找出最小的那条边<v,u>及点u，然后看看已属于T集合的有没有到达点u的距离和此最小边一样长，若有而且还不是点v，那么就证明有其他最小生成树.

#include<stdio.h>
#define MX 100000000
int cas,n,m,pre[110],dist[110],map[110][110];
int prim(int v0)
{
	for(int i=1;i<=n;i++)
	{
		pre[i]=v0;
		dist[i]=map[v0][i];
	}
	pre[v0]=-1;
	int sum=0;
	for(int i=1;i<n;i++)
	{
		int min=MX,u=-1;
		for(int j=1;j<=n;j++)
			if(pre[j]!=-1&&dist[j]<min)
			{
				min=dist[j];
				u=j;
			}
		for(int j=1;j<=n;j++)
			if(pre[j]==-1&&map[j][u]==min&&j!=pre[u])
			{
				return -1;
			}
		pre[u]=-1;
		sum+=min;
		for(int i=1;i<=n;i++)
			if(map[u][i]<dist[i]&&pre[i]!=-1)
			{
				pre[i]=u;
				dist[i]=map[u][i];
			}
	}
	return sum;
}
int main()
{
	scanf("%d",&cas);
	while(cas--)
	{
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				map[i][j]=MX;
		for(int i=1;i<=m;i++)
		{
			int a,b,c;
			scanf("%d%d%d",&a,&b,&c);
			map[a][b]=map[b][a]=c;
		}
		int k=prim(1);
		if(k==-1)
			printf("Not Unique!\n");
		else
			printf("%d\n",k);
	}
	return 0;
}