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

点击打开链接

判断最小生成树是否唯一！

用prime算法，其中然后加一步：假设枚举到M点如果 （（已经加入生成树中的点到M的权值等于未加入点M的权值）的个数）大于1，则肯定不是唯一！

#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;
}