poj1679The Unique MST

题目链接：http://poj.org/problem?id=1679

判断最小生成树是否唯一。

先求出最小生成树权值ans，用vector记录用的边。

依次不用vector里记录的边求生成树，若权值等于ans，说明不唯一。

#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;

int n,m;
struct edge
{
    int u,v,w;
    bool operator <(const edge &a) const {
    return w<a.w;
    }
}e[10010];
int f[110];
int ans;
vector<int>mst;

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

int gf(int x)
{
    return x==f[x]?x:f[x]=gf(f[x]);
}

void unit(int x,int y)
{
    int px=gf(x);
    int py=gf(y);
    f[px]=py;
}

void kru()
{
     init();
    for(int i=0;i<m;i++)
    {
        if(gf(e[i].u)!= gf(e[i].v))
        {
            mst.push_back(i);
            ans+=e[i].w;
            unit(e[i].u,e[i].v);
        }
    }

}
bool yesno()
{
    for(int i=0;i<mst.size();i++)
    {
        int cost=0;
          init();
          int k=0;
        for(int j=0;j<m;j++)
        {
            if(mst[i]==j) continue;
            if(gf(e[j].u)!=gf(e[j].v))
        {
            cost+=e[j].w;
            k++;
            unit(e[j].u,e[j].v);
        }
        }
        if(cost==ans&&k==n-1) return 1;
    }
    return 0;
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        mst.clear();
        ans=0;

        cin>>n>>m;

        for(int i=0;i<m;i++)
                cin>>e[i].u>>e[i].v>>e[i].w;

        sort(e,e+m);
        kru();
        if(yesno()) cout<<"Not Unique!
";
        else  cout<<ans<<endl;

    }
}