POJ 1679 The Unique MST Pim算法求解

题目链接：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66965#problem/K

题意：给定一个无向图，判断该图的最小生成树（MST)是否唯一。

思路：刚开始想的是学会Kruskal就好了，Prim算法没必要再写了吧，但是，这题明显用Prim简便了许多，所以有看了下Prim算法，算是对Prim彻底理解了。根据Prim求最小生成树的思路：每次加入一个点时，如果存在某个点到已知点集合S的最近距离不唯一时，那么，该最小生成树就不唯一，否则唯一。有了这个思路，在Prim算法中加入这个判断就好了。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define INF 0x7fffffff

using namespace std;

int N,M,Map[110][110];
int dis[110],vis[110];

void prim(){
    memset(vis,0,sizeof(vis));
    for(int i=0;i<=N;i++) dis[i]=Map[1][i];
    int sum=0;
    vis[1]=1;
    for(int p=0;p<N-1;p++){
        int Min=INF,t;
        for(int i=1;i<=N;i++) if(!vis[i] && dis[i]<=Min)
            Min=dis[i],t=i;
        int cnt=0;
        for(int i=1;i<=N;i++) if(vis[i] && Map[t][i]==Min){//判断该点的到S集合的最近距离是否唯一
            cnt++;
            if(cnt==2) {printf("Not Unique!\n");return;}
        }
        sum+=Min;vis[t]=1;
        for(int i=1;i<=N;i++) if(!vis[i] && dis[i]>Map[t][i])
            dis[i]=Map[t][i];
    }
    cout<<sum<<endl;
}

int main(){
    //freopen("D:\\in.txt","r",stdin);
    int T;cin>>T;
    while(T--){
        cin>>N>>M;
        for(int i=0;i<=N;i++) for(int j=0;j<=N;j++) Map[i][j]=INF;
        int a,b,c;
        for(int i=0;i<M;i++){
            scanf("%d %d %d",&a,&b,&c);
            Map[a][b]=Map[b][a]=c;
        }
        prim();
    }
    return 0;
}