POJ - 1679（次小生成树）

poj1679
分析：本题可以通过求出次小生成树，然后再比较次小生成树与最小生成是否相等来判断最小生成树是否唯一。
我们仔细回想一下kruskal求最小生成树的过程，最后得到的一定是一个无向连通图，如果我们要想求次小生成树的话，我们可以考虑枚举每一条不属于最小生成树的边，然后这条边会和原先的最小生成树的一些边构成一个环，在这个环内，我们需要找出一条最大的边：maxd[i][j] 来和那条边进行比较（没在最小生成树的的那条边：es[i]），此时我们要定义一个变量：pre=inf,假设变量sum是最小生成树的权值总和，那么通过pre=min(pre,sum+es[i].w-maxd[es[i].s][es[i].e])来更新pre的值，最后pre的值就是次小生成树的值。对于上述maxd[i][j]意思是：在最小生成树里从i到j的边中最大的边的权值，我们可以利用动态规划的思想来做：当集合要合并的时候我们就可以： maxd[G[px][j]][G[py][k]]=maxd[G[py][k]][G[px][j]]=es[i].w;
参考下面代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<algorithm>
#include<cmath>
#include<queue>
#include<cstring>
#include<vector>
using namespace std;
const int maxn=110;
const int inf=0x3f3f3f3f;
struct node{
    int s,e;
    int w;
    bool vis;
}es[maxn*maxn];
int p[maxn];
vector<int> G[maxn];
int maxd[maxn][maxn];
int n,m,sum;
bool cmp(node x,node y)
{
    return x.w<y.w;
}
int finds(int x)
{
    return p[x]==x?x:p[x]=finds(p[x]);
}
void kruskal()
{
    sort(es,es+m,cmp);
    for(int i=0;i<=n;i++){
        p[i]=i;
        G[i].clear();
        G[i].push_back(i);
    }
    int sum=0,k=0;
    for(int i=0;i<m;i++){
            if(k==n-1)break;
        int x=es[i].s;
        int y=es[i].e;
        int px=finds(x);
        int py=finds(y);
        if(px!=py){
            sum+=es[i].w;
            es[i].vis=true;
            k++;
           // if(k==n-1)break;//万万不能将这句话放在这，这句话一定要在下面两个for循环执行后再写
            for(int j=0;j<G[px].size();j++){
                for(int k=0;k<G[py].size();k++){
                    maxd[G[px][j]][G[py][k]]=maxd[G[py][k]][G[px][j]]=es[i].w;
                }
            }
            //if(k==n-1)break;
            p[px]=py;
            for(int j=0;j<G[px].size();j++){
                G[py].push_back(G[px][j]);
            }
        }
    }
    int pre=inf;
    for(int i=0;i<m;i++){
        if(!es[i].vis){
            pre=min(pre,sum+es[i].w-maxd[es[i].s][es[i].e]);
        }
    }
    //printf("%d %d\n",sum,pre);
    if(pre<=sum){
        printf("Not Unique!\n");
    }
    else{
        printf("%d\n",sum);
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&es[i].s,&es[i].e,&es[i].w);
            es[i].vis=false;
        }
        kruskal();
    }
    return 0;
}