您的位置:首页 > 产品设计 > UI/UE

poj1679(The Unique MST)Kruskal

2013-04-14 10:27 218 查看
 


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

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


 
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  poj