码迷,mamicode.com
首页 > 其他好文 > 详细

poj 1679 The Unique MST

时间:2018-02-17 16:40:09      阅读:179      评论:0      收藏:0      [点我收藏+]

标签:pac   click   release   uil   tracking   scanf   stat   ++   poj   

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

poj 1679 The Unique MST

标签:pac   click   release   uil   tracking   scanf   stat   ++   poj   

原文地址:https://www.cnblogs.com/zhchoutai/p/8451960.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!