POJ 1679 判断最小生成树是否唯一

题目链接： http://poj.org/problem?id=1679

题目大意：

　　给定一个n个点，m条边的无向图，(1<=n<=100，1<=m<=10000，因为保证没有重边)，问这个图的最小生成树（MST）是否唯一。

分析：

　　我的做法是第一次先用Prim算法做一次MST得到最终的权值total，并且标记所用的边（共n-1条），然后单独分别地枚举第一次做Prim时标记的边，对于枚举到的边，在原图里删除之，做一次MST看权值和第一次的total是否相等，如果相等说明MST不唯一，直到枚举完所有标记的边后还是没有判断到不唯一那么说明是唯一的，具体见代码，我觉得我的代码还是写得不错的，这得感谢fatboy_cw的头文件。

代码：

/*1679    Accepted    340K    16MS    C++    2170B    2012-04-19 11:17:19*/
#include <cstdio>
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

#define mpair make_pair
#define pii pair<int,int>
#define MM(a,b) memset(a,b,sizeof(a));
typedef long long lld;
typedef unsigned long long u64;
template<class T> bool up_max(T& a,const T& b){return b>a? a=b,1 : 0;}
template<class T> bool up_min(T& a,const T& b){return b<a? a=b,1 : 0;}
#define maxn 110
#define inf 100000000

int n,m;
int g[maxn][maxn], t[maxn][maxn];

int dis[maxn];
int pre[maxn];
bool vis[maxn];
bool Flag;
int Prim(){
    for(int i=1;i<=n;++i) dis[i]= inf, vis[i]= 0;
    for(int i=1;i<=n;++i){
        if( up_min( dis[i], t[i][1] ) )
            if( Flag ) pre[i]= 1;
    }

    vis[1]= 1;
    int ret= 0;
    for(int k=1;k<n;++k){
        int minval= inf, idx= -1;
        for(int i=1;i<=n;++i){
            if( !vis[i] && up_min( minval, dis[i] ) )
                idx= i;
        }
        if( -1 == idx ) return -1;
        vis[ idx ]= 1;
        ret+= dis[ idx ];
        for(int i=1;i<=n;++i){
            if( !vis[i] && up_min( dis[i], t[i][idx] ) )
                if( Flag ) pre[i]= idx;
        }
    }
    return ret;
}

void Copy(){
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j)
            t[i][j]= g[i][j];
}

int main()
{
    int T,u,v,w,i,j;
    cin>>T;
    while(T--){
        scanf("%d%d", &n, &m);
        for(i=1;i<=n;++i)
            for(j=1;j<=n;++j)
                g[i][j]= (i==j ? 0 : inf);

        for(i=1;i<=m;++i){
            scanf("%d%d%d", &u, &v, &w);
            g[u][v]= g[v][u]= w;
        }
        Copy();

        Flag= 1;
        int total= Prim();
        Flag= 0; ///

        if( -1 == total ){
            puts("0");
            continue; /// ???
        }

        for(i=2;i<=n;++i){
            Copy();
            t[ pre[i] ][i]= inf;
            t[i][ pre[i] ]= inf;
            int ret= Prim();
            if( ret==total ) break;
        }
        if( i>n ) printf("%d\n", total);
        else puts("Not Unique!");
    }
}