POJ 1679 The Unique MST（判断MST是否唯一）

题目链接：
POJ 1679 The Unique MST
题意：
判断最小生成树是否是唯一的。
分析：
最小生成树唯一：需要的边数最少并且总路径权值最小。
先生成一棵最小生成树，并把生成过程的各边记录下来，
然后依次去掉一条记录的边，再次生成最小生成树检查需要的边数和总路径权值和第一次的最小生成树是否完全一致。
如果完全一致，那么就最小生成树就不是唯一的，
否则每条记录的边都遍历过了都符合，那么就是唯一的。

//748K 16MS
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

const int maxn=110;
const int INF=0x3f3f3f3f;

int T,n,m,u,v,w,fu,fv,tot;
int pre[maxn],num[maxn];//num[i]标记生成第一次最小生成树时是否标记

struct Edge{
    int u,v,w;
}edge[maxn*maxn];

int find(int x)
{
    return pre[x]==x?x:pre[x]=find(pre[x]);
}

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

void init()
{
    for(int i=0;i<maxn;i++)
        pre[i]=i;
}

int main()
{
#ifdef  LOCAL
    freopen("in.txt","r",stdin);
#endif
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
        sort(edge,edge+m,cmp);
        init();
        int ans=0;
        tot=0;
        for(int i=0;i<m;i++)//第一次生成最小生成树
        {
            u=edge[i].u;
            v=edge[i].v;
            w=edge[i].w;
            fu=find(u);
            fv=find(v);
            if(fu!=fv)
            {
                num[tot++]=i;//记录各边
                pre[fu]=fv;
                ans+=w;//最小路径权值
            }
        }
        int flag=0;
        for(int i=0;i<tot;i++)
        {
            int tmpans=0;
            int tmptot=0;
            init();
            for(int j=0;j<m;j++)
            {
                if(num[i]==j) continue;//去掉这条记录边
                u=edge[j].u;
                v=edge[j].v;
                w=edge[j].w;
                fu=find(u);
                fv=find(v);
                if(fu!=fv)
                {
                    tmptot++;//去掉num[i]这条记录边生成最小生成树所需边数
                    pre[fu]=fv;
                    tmpans+=w;
                }
            }
            if(tmptot==tot&&tmpans==ans) flag=1;//完全一致
            if(flag) break;
        }
        if(flag) printf("Not Unique!\n");
        else printf("%d\n",ans);
    }
    return 0;
}