次小生成树

poj 1679 http://poj.org/problem?id=1679

次小生成树基于prim   o（v^2），可以通过次小生成树和最小生成树的值是否相等判断最小生成树是否唯一，若不等，则唯一，反之亦然。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define mt(a,b) memset(a,b,sizeof(a))
using namespace std;
const int inf=0x3f3f3f3f;
class SecondMinST_Prim{///次小生成树(无向图基于Prim)o(MV^2)
    typedef int typec;///边权的类型
    static const int MV=128;///点的个数
    int n,pre[MV];
    bool vis[MV],used[MV][MV];
    typec minst,cimin,cost[MV][MV],Max[MV][MV],lowc[MV];
    void Prim() {
        minst=0;
        mt(vis,0);
        mt(Max,0);
        mt(used,0);
        vis[0]=true;
        pre[0]=-1;
        for(int i=1; i<n; i++) {
            lowc[i]=cost[0][i];
            pre[i]=0;
        }
        lowc[0]=0;
        for(int i=1; i<n; i++) {
            typec sma=inf;
            int p=-1;
            for(int j=0; j<n; j++)
                if(!vis[j]&&sma>lowc[j]) {
                    sma=lowc[j];
                    p=j;
                }
            if(sma==inf){
                minst=-1;
                return ;
            }
            minst+=sma;
            vis[p]=true;
            used[p][pre[p]]=used[pre[p]][p]=true;
            for(int j=0; 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;
                }
            }
        }
    }
    void smst() {
        cimin=inf;
        for(int i=0; i<n; i++)
            for(int j=i+1; j<n; j++)
                if(cost[i][j]!=inf && !used[i][j]) {
                    cimin=min(cimin,minst+cost[i][j]-Max[i][j]);
                }
        if(cimin==inf) cimin=-1;
    }
public:
    void init(int tn){///传入点的个数，下标0开始
        n=tn;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                cost[i][j]=i==j?0:inf;
            }
        }
    }
    void add(int u,int v,typec w){
        if(cost[u][v]>w){
            cost[u][v]=w;
            cost[v][u]=w;
        }
    }
    void solve(){
        Prim();
        smst();
    }
    typec getmin(){
        return minst;
    }
    typec getci(){
        return cimin;
    }
}gx;
int main(){
    int t,n,m;
    while(~scanf("%d",&t)){
        while(t--){
            scanf("%d%d",&n,&m);
            gx.init(n);
            while(m--){
                int u,v,w;
                scanf("%d%d%d",&u,&v,&w);
                gx.add(u-1,v-1,w);
            }
            gx.solve();
            if(gx.getci()==gx.getmin()){
                puts("Not Unique!");
            }
            else{
                printf("%d
",gx.getmin());
            }
        }
    }
    return 0;
}

end