【POJ】 1679 The Unique MST(次小生成树的处理方法)

题目链接

题目大意：现在给你一个有N个点和M条边的一个图，然后就是问你这个图的最小生成树是不是唯一的，也就是说最小生成树的权值是不是能由多个生成树产生

思路：我对次小生成树的做法是这样的，先求出最小生成树，并且记录边，然后每次都去一条边，找生成树，看看是不是能产生一个和最小生成树一样的权值，就可以了

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=100+10;
int F[maxn];
vector<int >Edge;
int findset(int x)
{
    if(F[x]==-1)
        return x;
    return F[x]=findset(F[x]);
}
struct Node
{
    int u,v,dist;
    Node() {}
    Node(int a,int b,int c):u(a),v(b),dist(c) {}
} a[maxn*maxn];
int cmp(Node a,Node b)
{
    return a.dist<b.dist;
}
int n,m;
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        int cnt=0;
        for(int i=1; i<=m; i++)
        {
            int u,v,dist;
            scanf("%d%d%d",&u,&v,&dist);
            a[i]=Node(u,v,dist);
        }
        memset(F,-1,sizeof(F));
        Edge.clear();
        sort(a+1,a+1+m,cmp);
        int ans1=0;
        int i=1;
        for(int j=1; i<=n-1&&j<=m; j++)
        {
            Node node=a[j];
            int u=node.u,v=node.v,dist=node.dist;
            int fu=findset(u),fv=findset(v);
            if(fu!=fv)
            {
                Edge.push_back(j);
                ans1+=dist;
                F[fu]=fv;
                i++;
            }
        }
        if(i!=n) ans1=100000000+9999999;
        int ans2=100000000+9999999;
        int temp=0;
        for(int s=0; s<Edge.size(); s++)
        {
            memset(F,-1,sizeof(F));
            temp=0;
            int jump=Edge[s];
            //cout<<"jump:"<<jump<<endl;
            int num=1;
            for(int j=1;num<=n-1&&j<=m; j++)
            {
                if(j==jump)continue;
                Node node=a[j];
                int u=node.u,v=node.v,dist=node.dist;
                int fu=findset(u),fv=findset(v);
                if(fu!=fv)
                {
                    temp+=dist;
                    F[fu]=fv;
                    num++;
                }
            }
            if(num==n)
            {
                ans2=min(ans2,temp);
            }
        }
        if(ans2==ans1)
            printf("Not Unique!\n");
        else
            printf("%d\n",ans1);
    }
}