poj 1679 求最小生成树是否唯一

点击打开链接
//最小生成树是否唯一
#include <iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int maxv=100+5;
const int maxe=5000+5;
int parent[maxv];
struct Edge{
int from,to,weight;
};
Edge edges[maxe];
bool operator<(const Edge &E1,const Edge &E2)
{
    return E1.weight<E2.weight;
}
int find(int x)
{
    return x==parent[x]?x:parent[x]=find(parent[x]);
}
void init(int v)
{
    for(int i=1;i<=v;i++)
        parent[i]=i;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int v,e;
        scanf("%d%d",&v,&e);
        init(v);
        for(int i=0;i<e;i++)
        {
            int v1,v2,dist;
            scanf("%d%d%d",&v1,&v2,&dist);
            edges[i].from=v1;
            edges[i].to=v2;
            edges[i].weight=dist;
        }
        sort(edges,edges+e);
        //记录最小生成树中的各条边，并依次去掉进行枚举
        int sum=0;      //最小花费
        vector<int>str;
        for(int i=0;i<e;i++)
        {
            int p1=find(edges[i].from);
            int p2=find(edges[i].to);
            if(p1==p2) continue;
            parent[p1]=p2;
            sum+=edges[i].weight;
            str.push_back(i);
        }
        int k=str.size();
        bool flag=false;
        for(int i=0;i<k;i++)
        {
            int sss=0;
            int ss=0;
            int f=str[i];
            init(v);
            for(int j=0;j<e;j++)
            {
                int p1=find(edges[j].from);
                int p2=find(edges[j].to);
                if(p1==p2||j==f) continue;
                parent[p1]=p2;
                sss+=edges[j].weight;
                ss++;
            }
            if(ss==v-1&&sss==sum) flag=true;            //坑点：有时候并不能构成一棵最小生成树，需要进行判断
            if(flag) break;
        }
        if(!flag) printf("%d\n",sum);
        else printf("Not Unique!\n");
        str.clear();
    }
    return 0;
}