【POJ 1679】The Unique MST（次小生成树）

找出最小生成树，同时用Max[i][j]记录i到j的唯一路径上最大边权。然后用不在最小生成树里的边i-j来替换，看看是否差值为0。

#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int N=;
const int INF=0x3f3f3f3f;
int n,m,ans,s;
int lowc[N],vis[N],g[N][N];
int Max[N][N],fa[N],used[N][N];
int Prim(){
    int ans=;
    memset(Max,,sizeof Max);
    memset(used,,sizeof used);
    memset(vis,,sizeof vis);
    vis[]=;
    for(int i=;i<n;i++){
        lowc[i]=g[][i];fa[i]=;
    }
    for(int i=;i<n;i++){
        int p=;
        while(vis[p])p++;
        for(int j=p+;j<n;j++)
            if(!vis[j] && lowc[j]<lowc[p])
                p=j;
        if(lowc[p]==INF)return -;//原图不连通
        ans+=lowc[p];
        vis[p]=;
        used[p][fa[p]]=used[fa[p]][p]=;
        for(int j=;j<n;j++)
            if(vis[j])
                Max[j][p]=Max[p][j]=max(Max[j][fa[p]],lowc[p]);                    else if(g[p][j]<lowc[j]){
                lowc[j]=g[p][j];
                fa[j]=p;
            }
    }
    return ans;
}
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        memset(g,INF,sizeof g);
        scanf("%d%d",&n,&m);
        while(m--){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            u--;v--;
            g[u][v]=g[v][u]=w;
        }
        ans=Prim();
        s=INF;
        for(int i=;i<n;i++)
            for(int j=i+;j<n;j++)
                if(g[i][j]!=INF&&!used[i][j])
                    s=min(s,g[i][j]-Max[i][j]);

        if(ans==-||s==)
            puts("Not Unique!");
        else
            printf("%d\n",ans);
    }
    return ;
}

　　wa了好几发，原因是，s初始化为ans，而如果ans本身就是0的话，应该是唯一的最小生成树。