poj 1679 The Unique MST (次小生成树)

首先说明这道题图式连通的.
    判断最小生成树是否唯一等价于判断最小生成树与次小生成树是否相等。
    题目就清晰了。

    次小生成树不清楚的 看看我的另一篇 hdu 4081 Qin Shi Huang's National Road System(次小生成树)

#include <stdio.h>
#include <string.h>
#include<math.h>
#define MAX 105
#define MAXCOST 0x7fffffff
#define max(a,b) a>b?a:b
int n,m;
int graph[MAX][MAX],sum,min;
int dp[MAX][MAX];
bool adjvex[MAX];//记录顶点是否被访问
int lowcost[MAX];//只管记录的是已是最小生成树集合到各个点的最小值
int pre[MAX],p;//最小生成树中的点的父节点
void Prim()
{
    int i,j,k;
    for(i=1;i<=n;i++)
    {
        lowcost[i]=graph[1][i];   //从第一个的顶点开始
        pre[i]=1;
    }
    memset(dp,0,sizeof(dp));
    memset(adjvex,false,sizeof(adjvex));
    adjvex[1]=true;
    for(i=2;i<=n;i++)//这里注意因为第一个点在循环外算了。所以还需进行n-1次
    {
        min=MAXCOST;
        for(j=2;j<=n;j++)//选出当前最小生成树集合到非集合点的最小的边
            if(adjvex[j]==false &&lowcost[j]!=-1&& lowcost[j]<min)
            {
                min=lowcost[j];
                k=j;    //记下最小的点
            }
        p=pre[k];
        dp[k][p]=dp[p][k]=graph[p][k];
        for(j=1;j<=n;j++)//更新k j上最大的边。
        {
            if(adjvex[j])
            {
                    dp[k][j]=dp[j][k]=max(dp[j][p],dp[p][k]);//k是刚刚加入生成树的点.他与其他点的联系都父节点建立.
            }
        }
        adjvex[k]=true;   //为true表示该权值已经是最小，为flase是还不确定，应继续更新
        for(j=2;j<=n;j++)
        {
            if(adjvex[j]==false && graph[k][j]!=-1&&(lowcost[j]==-1||lowcost[j]>graph[k][j]))  //更新lowcost
             {
                 pre[j]=k;
                 lowcost[j]=graph[k][j];
             }
        }
    }
    for(i=2;i<=n;i++)//第一个点早选了。从2开始
    {
            sum+=lowcost[i];  //求最小生成树路径长度
    }
}
bool getAns()//遍历每一对点
{
    for(int i=1;i<n;i++)
    {
        for(int j=i+1;j<=n;j++)
        {
                    if(pre[i]!=j&&pre[j]!=i&&graph[i][j]!=-1)//保证i j不是最小生成树的点。并且是连通的。
                    {
                        if(graph[i][j]==dp[i][j])
                            return false;
                    }
        }
    }
    return true;
}
int main()
{
    freopen("test.txt","r",stdin);
    int x,y,z,T;
    scanf("%d",&T);
    while(T--)
    {
        memset(graph,-1,sizeof(graph));
        scanf("%d%d",&n,&m);
        sum=0;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
                graph[x][y]=z;
                graph[y][x]=z;
        }
        Prim();
        if(getAns())
            printf("%d\n",sum);
        else
        {
            printf("Not Unique!\n");
        }
    }
}