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

poj 1679 The Unique MST

2015-07-21 13:40 288 查看
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct data
{
int u,v,w;
int same;
int use;
int del;
}e[10000+5];

int n,m,bin[1000+5],f;

bool cmp(data a,data b)
{
return a.w<b.w;
}

int Find(int x)
{
int s;
for(s=x;bin[s]>=0;s=bin[s]);

while(s!=x)
{
int t=bin[x];
bin[x]=s;
x=t;
}
return s;
}

void Union(int x1,int x2)
{
int f1=Find(x1),f2=Find(x2);
int t=bin[f1]+bin[f2];
if(bin[f1]>bin[f2])
{
bin[f1]=f2;
bin[f2]=t;
}
else
{
bin[f2]=f1;
bin[f1]=t;
}

}

int kruskal()
{
int i,num=0,u,v,ans=0;
for(i=0;i<=n+5;i++)
bin[i]=-1;

for(i=0;i<m;i++)
{
//printf("%d\n",i);
if(e[i].del==1) continue;
u=e[i].u;
v=e[i].v;
if(Find(u)!=Find(v))
{
Union(u,v);
num++;
ans+=e[i].w;
if(f) e[i].use=1;
}
if(num>=n-1) break;
}
return ans;
}

int main()
{
int _,i,j,k,ans1,ans2;
scanf("%d",&_);
while(_--)
{
scanf("%d%d",&n,&m);
for(i=0;i<m;i++)
{
scanf("%d %d %d",&e[i].u,&e[i].v,&e[i].w);
e[i].u--;
e[i].v--;
e[i].use=0;
e[i].same=0;
e[i].del=0;
}

for(i=0;i<m;i++)
for(j=0;j<m;j++)
{
if(i==j) continue;
if(e[i].w==e[j].w)
e[i].same=e[j].same=1;
}
f=1;
sort(e,e+m,cmp);
ans1=kruskal();
f=0;
for(i=0;i<m;i++)
{
if(e[i].same==1&&e[i].use==1)
{
e[i].del=1;
ans2=kruskal();
if(ans2==ans1)
{
printf("Not Unique!\n");
break;
}
e[i].del=0;
}
}

if(i>=m)
printf("%d\n",ans1);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: