次小生成树题（k） poj1679The Unique MST

http://poj.org/problem?id=1679

#include<iostream>

#include<vector>

#include<cstdio>

#include<cstring>

#include<algorithm>

using namespace std;

struct node{

    int u,v,w;

}e[];

int f[];

vector<int>vis;

bool cmp(node p,node q){

    return p.w<q.w;

}

int find(int x){

    return x==f[x]?x:f[x]=find(f[x]);

}

int main(){

    int t;

    scanf("%d",&t);

    while(t--){

        int n,m,ans=;

        vis.clear();

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

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

            int u,v,w;

            scanf("%d%d%d",&u,&v,&w);

            e[i].u=u;

            e[i].v=v;

            e[i].w=w;

        }

        sort(e+,e++m,cmp);

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

            f[i]=i;

        for(int j=,i=;i<=m;i++){

            int u=e[i].u,v=e[i].v;

            int a=find(u),b=find(v);

            if(a!=b){

                f[a]=b;

                j++;

                ans+=e[i].w;

                vis.push_back(i);

            }

            if(j==n-)

                break;

        }

        int flag=;

        for(int k=;k<vis.size();k++){

            int sign=vis[k];

            int ans1=;

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

                f[i]=i;

            int j=;

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

                if(i!=sign){

                    int u=e[i].u,v=e[i].v;

                    int a=find(u),b=find(v);

                    if(a!=b){

                        f[a]=b;

                        ans1+=e[i].w;

                        j++;

                    }

                    if(j==n-)

                        break;

                }

            }

            if(j==n-){

                if(ans1==ans){

                    flag=;

                    break;

                }

            }

        }

        if(flag)

            puts("Not Unique!");

        else

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

    }

    return ;

}