poj1679(判断最小生成树是否唯一)

题意：给出n个点,m条边，要你判断最小生成树是否唯一。

思路：先做一次最小生成树操作，标记选择了的边，然后枚举已经被标记了的边，判断剩下的边组成的最小生成树是否与前面的相等，相等，则不唯一，否则唯一......

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

using namespace std;

struct node

{

    int v1,v2;

    int dis;

    int flg;

} s[10000];



int cmp(const node a,const node b)

{

    if(a.dis<b.dis)

        return 1;

    else

        return 0;

}



int father[200],n;



int kruskal(int num,int m)

{

    int ans=0,cnt=1;

    for(int i=0; i<m; i++)

    {

        if(i==num)

            continue;

        int s1=father[s[i].v1];

        int s2=father[s[i].v2];

        if(s1!=s2)

        {

            ans+=s[i].dis;

            //s[i].flg=1;

            cnt++;

            father[s2]=s1;

            for(int j=0; j<=n; j++)

                if(father[j]==s2)

                    father[j]=s1;

        }

    }

    if(cnt!=n)  return -1;

    else   return ans;

}

int main()

{

    int text;

    scanf("%d",&text);

    while(text--)

    {

        int m;

        int ans=0;

        scanf("%d%d",&n,&m);

        for(int i=0; i<=n; i++)

            father[i]=i;

        for(int i=0; i<m; i++)

        {

            scanf("%d%d%d",&s[i].v1,&s[i].v2,&s[i].dis);

            s[i].flg=0;

        }



        sort(s,s+m,cmp);

        int cnt=1;

        for(int i=0; i<m; i++)

        {

            int s1=father[s[i].v1];

            int s2=father[s[i].v2];

            if(s1!=s2)

            {

                ans+=s[i].dis;

                s[i].flg=1;

                cnt++;

                father[s2]=s1;

                for(int j=0; j<=n; j++)

                    if(father[j]==s2)

                        father[j]=s1;

            }

        }

        int w=0;

        for(int i=0; i<m; i++)

        {

            if(s[i].flg==0)

                continue;

            int sum=0;

            for(int j=0;j<=n;j++)

            father[j]=j;

            sum=kruskal(i,m);

            //printf("sum==%d\n",sum);

            if(sum==ans)

            {

                w=1;

                break;

            }

        }

        if(w==1)

            printf("Not Unique!\n");

        else

            printf("%d\n",ans);

    }

    return 0;

}