次小生成树

今天我复习了一下次小生成树。
首先介绍一下最小生成树：
最小生成树表示在一个图中，你找到一颗树（包含所有的顶点），该树满足对于任意的x与y（x！=y）x到y的路径最小。
然后怎么求最小生成树呢？
有两种贪心方法：
kruskal算法
对于所有边（E集合）我们排一遍序，在建另外一个图D（在这个图上只有n个点无边）。
从小到大依次取出边，看这条边的两端点是否在D上连通，若连通，则不加入这条边；否则，在D图上加入这条边。
如果所有边都在一个连通块内则退出。
总复杂度为O（nlogn）；
Prim算法
首先我们有一个空队列a，先讲1号点加入，x=1。
对于队列中的元素x，先将它作标记d[x]=1（表示曾经出现过）。
在用x对于它所相连的顶点u的dist值进行修改，若d[u]!=1 && dist[u]>dist[x]+f[x][u];则dist[u]=dist[x]+f[x][u];
(dist[u]表示任何一个点（除u外）到u的距离且距离最小,d[u]=1表示u点曾经出现过)
在把dist最小且没出现过的点选出来使它为下一个x，当选不出来x时，退出。
总复杂度为O（n^2）。
次小生成树其实就是第二小生成树吧。。。
下面就是怎么求这玩意？
如果有一天非树边加到树上必定会形成环，假如这个环上的一条原树边=这条边，则最小生成树不是唯一的。
因此我们需要枚举所有的非树边。
max[i][j]表示在树上i点到j点形成环的最大边距离。
（即假如i到j有一条路径且f[i][j]!=0则可以形成环，则需要维护这个环上的最大树边）
假设我们已经求出了max[i][j],则枚举顶点k
max[j][k]=max0(max[fa][k],max[j][k]);（fa是k的父亲）
只可以用prime算法，因为你加入一条边时，这颗树是要连通的，否则k无父亲。
最后送上代码：

#include <stdio.h>
#include <math.h>
#include <string.h>
#define rep(i,l,r) for(i=l;i<=r;i++)
#define INF 100456
int n,m,t;
int f[300][500];
int max[500][500];
int d[500][500];
int max0(int a,int b)
{
    if(a>=b)return a;
    else return a;
}
int prime()
{
    int i,j,k,min,fa,ans=0;
    int dist[600],pre[600],funk[600];
    rep(i,1,n){
        dist[i]=f[1][i];
        pre[i]=1;
        funk[i]=0;
    }
    dist[1]=0;pre[1]=0;funk[1]=1;
    rep(i,1,n){
        min=INF;k=-1;
        rep(j,1,n)
         if(funk[j]!=1 && dist[j]<min){
            min=dist[j];k=j;
         }
        if(k==-1)break;
        funk[k]=1;fa=pre[k];
        max[fa][k]=dist[k];max[k][fa]=dist[k];
        d[k][fa]=1;d[fa][k]=1;
        ans+=dist[k];
        rep(j,1,n)
         max[j][k]=max0(max[fa][k],max[j][k]);
        rep(j,1,n)
         if( funk[j]!=1 && dist[j]>f[k][j] && f[k][j]!=0){
             dist[j]=f[k][j];   
             pre[j]=k;
         }
    }
    return ans;
}
int work()
{
    int i,j,k=0,p;
    p=prime();
    rep(i,1,n)
        rep(j,1,n){
            if(d[i][j]!=0 || f[i][j]==INF)continue;
            if(f[i][j]==max[i][j]){
                k=1;break;
            }
        }       
    if(k==1)printf("Not Unique!\n");   
    else printf("%d\n",p);
    return 0;
}
int init()
{
    int i,j,k,x,y,z;
    scanf("%d",&t);
    while(t>0){
        memset(max,0,sizeof(max));
        memset(f,INF,sizeof(f));
        memset(d,0,sizeof(d));
        scanf("%d%d",&n,&m);
        rep(i,1,m){
            scanf("%d%d%d",&x,&y,&z);
            f[x][y]=z;f[y][x]=z;
        }
        work();
        t--;
    }
    return 0;
}
int main()
{
    init();
    //work();
    return 0;
}

其实这也是poj1679原题。。。