poj1679.The Unique MST（最小生成树是否唯一）

题意:判断最小生成树是否唯一
转化为次小生成树做法

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=110;
const int INF=0x3f3f3f3f;
int vis[maxn];
int ans;
int lowc[maxn];
int pre[maxn];
int Max[maxn][maxn];//Max[i][j]表示在最小生成树中从i到j的路径中的最大边权
bool used[maxn][maxn];
int cost[maxn][maxn];

void prim(int n){
    ans=0;
    memset(vis,0,sizeof(vis));
    memset(Max,0,sizeof(Max));
    memset(used,false,sizeof(used));
    vis[1]=1;
    pre[1]=-1;
    for(int i=2;i<=n;i++){
        lowc[i]=cost[1][i];
        pre[i]=1;
    }
    lowc[1]=0;
    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 ;
        ans+=minc;
        vis[p]=1;
        used[p][pre[p]]=used[pre[p]][p]=true;
        for(int j=1;j<=n;j++){
            if(vis[j])
                Max[j][p]=Max[p][j]=max(Max[j][pre[p]],lowc[p]);
            if(!vis[j]&&lowc[j]>cost[p][j]){
                lowc[j]=cost[p][j];
                pre[j]=p;
            }
        }
    }
}

int main(){
    int t,n,m;
    int a[1101],b[1110],c[1101];
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        int k;
        memset(cost,INF,sizeof(cost));
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&a[i],&b[i],&k);
            c[i]=k;
            cost[a[i]][b[i]]=cost[b[i]][a[i]]=min(cost[a[i]][b[i]],k);
        }
        prim(n);
        int flag=0;
        for(int i=1;i<=n;i++){
            if(used[a[i]][b[i]])
                continue;
            else{
                if(c[i]==Max[a[i]][b[i]]){
                    flag=1;
                    break;
                }
            }
        }
        if(flag==1){
            printf("Not Unique!\n");
        }
        else
            printf("%d\n",ans);
    }
    return 0;
}