UVA10462-Is There A Second Way Left?

题意：有n个点，m条边，求最小生成树

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>
#include <set>
#include <stack>
#include <map>
#include <climits>

using namespace std;

#define LL long long
const int INF=0x3f3f3f3f;

#define MAXV 110
#define MAXE 10100

struct node
{
    int s,e,w,flag;
}x[MAXE];

int n,m;
int f[MAXV];

int cmp(node a,node b)
{
    return a.w<b.w;
}

int Find(int x)
{
    if(f[x]==x) return x;
    else return f[x]=Find(f[x]);
}

bool Union(int a,int b)
{
    int aa=Find(a);
    int bb=Find(b);
    if(aa==bb) return 0;
    f[aa]=bb;
    return 1;
}

int kruskal()
{
    int sum=0,cnt=0;
    for(int i=0; i<=n; i++) f[i]=i;
    for(int i=0; i<m; i++)
    {
        if(Union(x[i].s,x[i].e))
        {
            sum+=x[i].w;
            x[i].flag=1;
            cnt++;
            if(cnt==n-1) break;
        }
    }
    if(cnt==n-1) return sum;
    return -1;
}

int kruskalt()
{
    int sum=0,cnt=0;
    for(int i=0;i<=n;i++) f[i]=i;
    for(int i=0;i<m;i++)
    {
        if(Union(x[i].s,x[i].e))
        {
            sum+=x[i].w;
            cnt++;
            if(cnt==n-1) break;
        }
    }
    if(cnt==n-1) return sum;
    return -1;
}

int main()
{
    int t,cas=0;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&x[i].s,&x[i].e,&x[i].w);
            x[i].flag=0;
        }
        sort(x,x+m,cmp);
        int ans=kruskal();
        printf("Case #%d : ",++cas);
        if(ans==-1) {printf("No way\n");continue;}
        int flag=0,ans2=INF;
        for(int i=0; i<m; i++)
        {
            if(x[i].flag)
            {
                int s=x[i].s;
                x[i].s=x[i].e;
                int tmp=kruskalt();
                if(tmp!=-1) ans2=min(ans2,tmp);
                x[i].s=s;
            }
        }
        if(ans2==INF) printf("No second way\n");
        else printf("%d\n",ans2);
    }
    return 0;
}