今天我复习了一下次小生成树。
首先介绍一下最小生成树:
最小生成树表示在一个图中,你找到一颗树(包含所有的顶点),该树满足对于任意的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原题。。。