[ An Ac a Day ^_^ ][kuangbin带你飞]专题八 生成树 POJ 1679 The Unique MST

求最小生成树是否唯一

求一遍最小生成树再求一遍次小生成树 看看值是否相等就可以

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#define cl(a,b) memset(a,b,sizeof(a))
#define debug(x) cerr<<#x<<"=="<<(x)<<endl
using namespace std;
typedef long long ll;

const int maxn=1e2+10;
const int inf=0x3f3f3f3f;

int n,m;
int cost[maxn][maxn];
int pre[maxn],lowc[maxn];
int Max[maxn][maxn];
bool vis[maxn],used[maxn][maxn];

int mst()
{
    int ans=0;
    cl(vis,false),cl(Max,0),cl(used,false);
    vis[1]=true,pre[1]=-1,lowc[1]=0;
    for(int i=2;i<=n;i++)
    {
        lowc[i]=cost[1][i];
        pre[i]=1;
    }
    for(int i=2;i<=n;i++)
    {
        int minc=inf;
        int p=-1;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&minc>lowc[j])
            {
                minc=lowc[j];
                p=j;
            }
        }
//        if(minc==inf) return -1;
        ans+=minc;
        vis[p]=used[p][pre[p]]=used[pre[p]][p]=true;
        for(int j=1;j<=n;j++)
        {
            if(vis[j])
            {
                Max[j][p]=max(Max[j][pre[p]],lowc[p]);
            }
            if(!vis[j]&&lowc[j]>cost[p][j])
            {
                lowc[j]=cost[p][j];
                pre[j]=p;
            }
        }
    }
    return ans;
}

void solve()
{
    int ans=mst();
    int Min=inf;
    for(int i=1;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
        {
            if(cost[i][j]!=inf&&!used[i][j])
            {
                Min=min(Min,ans+cost[i][j]-Max[i][j]);
            }
        }
    }
    if(ans==Min) printf("Not Unique!
");
    else printf("%d
",ans);
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        cl(cost,inf);
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            cost[u][v]=cost[v][u]=w;
        }
        solve();
    }
    return 0;
}