K - The Unique MST - poj 1679

#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;

const int maxn = 105;
const int oo = 0x7fffffff;

int G[maxn][maxn], path[maxn][maxn];//path记录两点之间的最大边值
bool use[maxn][maxn];//记录是否是最小生成树上面的边

int Prim(int N)
{
    int dist[maxn], vis[maxn]={0,1}, pre[maxn];
    int i, ans = 0, T = N-1;

    for(i=1; i<=N; i++)
    {
        pre[i] = 1;
        dist[i] = G[1][i];
    }
    while(T--)
    {
        int k = 1, mini = oo;

        for(i=1; i<=N; i++)
        {
            if(!vis[i] && mini > dist[i])
                mini = dist[i], k = i;
        }
        use[ pre[k] ][k] = use[k][ pre[k] ] = true;
        ans += mini; vis[k] = true;

        for(i=1; i<=N; i++)
        {
            if(vis[i] && k != i)
                path[k][i]=path[i][k] = max(path[pre[i]][i], mini);
            if(!vis[i] && dist[i] > G[k][i])
                dist[i] = G[k][i], pre[i] = k;
        }
    }

    return ans;
}
int OK(int N)//判断是否有次小生成树
{
    for(int i=1; i<=N; i++)
    for(int j=i+1; j<=N; j++)
    {
        //如果有边的长度与最小树上的边相等，就说明不唯一了
        if(!use[i][j] && path[i][j]==G[i][j])
            return 0;
    }

    return 1;
}

int main()
{
    int T;

    scanf("%d", &T);

    while(T--)
    {
        int i, j, N, M, u, v, w;

        scanf("%d%d", &N, &M);

        for(i=1; i<=N; i++)
        for(j=1; j<=N; j++)
        {
            G[i][j] = (i == j ? 0 : oo);
            path[i][j] = 0;
            use[i][j] = false;
        }

        while(M--)
        {
            scanf("%d%d%d", &u, &v, &w);
            G[u][v] = G[v][u] = w;
        }

        int ans = Prim(N);

        if(OK(N) == 0)
            printf("Not Unique!\n");
        else
            printf("%d\n", ans);
    }