次小生成树题（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 ;
}