阅读 62

poj1679(The Unique MST)Kruskal


最近略忙,就不写题意思路什么的,直接上代码。

#include<stdio.h>
#include<stdlib.h>
struct edge
{
	int u,v,w,flag;
}p[4952];
int n,m;
int f[101];
int used[101];
int cmp(const void*aa,const void*bb)
{
	return ((struct edge*)aa)->w-((struct edge*)bb)->w;
}
int find(int x)
{
	return f[x]==x?x:(f[x]=find(f[x]));
}
int Kruskal()
{
	int sum=0,i,x,y,t=0;
	for(i=0;i<m;i++)
	{
		x=find(p[i].u);
		y=find(p[i].v);
		if(x!=y)
		{
			f[x]=y;
			sum+=p[i].w;
			used[t]=i;
			t++;
			if(t==n-1) break;
		}
	}
	return sum;
}
int reKruskal()
{
	int sum=0,i,x,y,t=0;
	for(i=0;i<m;i++)
	{
		x=find(p[i].u);
		y=find(p[i].v);
		if(x!=y&&!p[i].flag)
		{
			f[x]=y;
			sum+=p[i].w;
			t++;
			if(t==n-1) break;
		}
	}
	return sum;
}
int main()
{
	//freopen("12.3.4.input.txt","r",stdin);
	int t,i,j,ans,tans,k,pt=0;
	scanf("%d",&t);
	for(i=0;i<t;i++)
	{
		scanf("%d %d",&n,&m);
		for(j=1;j<=n;j++) f[j]=j;
		for(j=0;j<n;j++) used[j]=-1;
		for(j=0;j<m;j++)
		{
			scanf("%d %d %d",&p[j].u,&p[j].v,&p[j].w);
			p[j].flag=0;
		}
		qsort(p,m,sizeof(p[0]),cmp);
		ans=Kruskal();
		pt=0;
		for(j=0;j<n-1;j++)
		{
			p[used[j]].flag=1;
			for(k=1;k<=n;k++) f[k]=k;
			tans=reKruskal();
            p[used[j]].flag=0;
			if(ans==tans&&ans!=0) 
			{
				pt=1;
				break;
			}
		}
		if(pt) printf("Not Unique!\n");
		else printf("%d\n",ans);
	}
	return 0;
}


文章分类
代码人生
版权声明:本站是系统测试站点,无实际运营。本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 XXXXXXo@163.com 举报,一经查实,本站将立刻删除。
相关推荐