今天我们就来浅谈一下最小树形图的问题。
大题上完整的朱、刘算法是由四个大步骤组成的:
1、求最短弧集合E
2、判断集合E中有没有有向环,如果有转步骤3,否则转4
3、收缩点,把有向环收缩成一个点,并且对图重新构建,包括边权值的改变和点的处理,之后再转步骤1。
4、展开收缩点,求得最小树形图。
因为我们ACM一般情况下都是在考察队最小树型图的权值问题,所以一般省略步骤4,对于其环的权值和在中间处理过程中就可以处理完毕。所以我们这里就不多讨论第四个点了。
我们分步处理、
1、首先我们先求最短弧集合E,对于当前图如果有n个点(一个有向环的收缩点算作一个点),我们就要选出n-1个点,确定其入边的最短边,由其组成的一个集合我们就叫做最短弧集合E,如果我们枚举到某一个点的时候,它没有入边,那么说明不存在最小树形图,所以这个时候算法结束,回到主函数。
代码实现:
for(int i=1; i<=n; i++)if(i!=u&&!flag[i])//u作为图的根节点,flag【i】为1的情况就是表示这个点在某个有向环里边,并且他不是这个有向环的代表点(缩点)
{
w[i][i]=INF, pre[i] = i;//首先让当前点的前驱点为自己。
for(int j=1; j<=n; j++)if(!flag[j] && w[j][i]<w[pre[i]][i])//枚举i的前驱点(从j能够到i的点),并且求其最短边,加入集合E中
{
pre[i] = j;//并且标记当前点的前驱点为j
}
if(pre[i]==i)return -1;//如果当前枚举到的点i没有入边,那么就不存在最小树形图(因为一颗树是要所有节点都是连通的啊)
}
int i;
for(i=1; i<=n; i++)
{
if(i!=u&&!flag[i])
{
int j=i, cnt=0;
while(j!=u && pre[j]!=i && cnt<=n) j=pre[j], ++cnt;//对于每个节点都找前驱节点,看看能否成环。
if(j==u || cnt>n) continue; //最后能找到起点(根)或者是走过的点已经超过了n个,表示没有有向环
break;//表示有有向环
}
}
上图变换成语言描述:如果点u在环内,如果点k在环外,并且从k到u有一条边map【u】【v】=w,并且在环内还有一点i,使得map【i】【k】=w2,辣么map【k】【收缩点】=w-w2;
基于贪心思想,对于环的收缩点i和另外一点k(也在环内),对于环外一点j,如果map【k】【j】 <map【i】【j】,辣么map【i】【j】=map【k】【j】,因为是有向图,入收缩点的边要这样处理,出收缩点的边也要这样处理,对于刚刚三个步骤:收缩点,收缩点处理新图的边权值,以及基于贪心思想的最终处理点i的出边入边权值,后两者我们可以合并成一个操作,其分别的代码实现:
3、收缩点:
int j=i;
memset(vis, 0, sizeof(vis));
do
{
ans += w[pre[j]][j], j=pre[j], vis[j]=flag[j]=true;//对环内的点标记,并且直接对环的权值进行加和记录,在最后找到最小树形图之后就不用展开收缩点了
}
while(j!=i);
flag[i] = false; // 环缩成了点i,点i仍然存在
4、处理收缩点后形成的图:
for(int k=1; k<=n; ++k)if(vis[k]) // 在环中点点,刚刚在收缩点的时候,已经把在环中的点进行标记了。
{
for(int j=1; j<=n; j++)if(!vis[j]) // 不在环中的点
{
if(w[i][j] > w[k][j]) w[i][j] = w[k][j];
if(w[j][k]<INF && w[j][k]-w[pre[k]][k] < w[j][i])
w[j][i] = w[j][k] - w[pre[k]][k];
}
}
处理完4之后,我们就回到步骤1,继续找最小弧集E,最后找到了一个没有环的最小弧集E之后,对于没有弧的集合E中的所有边(包括能将收缩点展开的边)就是我们要求的最小树形图的边集。
因为我们ACM一般求的都是最小树形图的权值,所以我们一般不需要展开收缩点,在处理环的时候,直接将其边权值记录下来就好,当找到一个没有环的集合E的时候,对其中的最后边权值进行加和即可,对于最后这部分的加权,代码实现:
if(i>n)//这块代码是紧接着代码2之后的部分,如果枚举了所有点i都没有发现有向环,辣么就是找到了这个最终集合。
{
for(int i=1; i<=n; i++)if(i!=u && !flag[i]) ans+=w[pre[i]][i];//最后对这个最后的集合E里边所有边加和即可。
return ans;
}
完整的朱、刘算法代码实现(没有展开收缩点的):
void init()//不能少了初始化的内容
{
memset(vis, 0, sizeof(vis));
memset(flag, 0, sizeof(flag));
for(int i=0; i<=n; i++)
{
w[i][i] = INF;
for(int j=i+1; j<=n; j++)
w[i][j]=w[j][i]=INF;
}
}
double directed_mst(int u)//u表示根节点
{
double ans=0;
memset(vis, 0, sizeof(vis));
while(true)
{
//求最短弧集合E
for(int i=1; i<=n; i++)if(i!=u&&!flag[i])
{
w[i][i]=INF, pre[i] = i;
for(int j=1; j<=n; j++)if(!flag[j] && w[j][i]<w[pre[i]][i])
{
pre[i] = j;
}
if(pre[i]==i)return -1;//也可以用dfs预处理判断凸的连通
}
//判断E是否有环
int i;
for(i=1; i<=n; i++)
{
if(i!=u&&!flag[i])
{
int j=i, cnt=0;
while(j!=u && pre[j]!=i && cnt<=n) j=pre[j], ++cnt;
if(j==u || cnt>n) continue; //最后能找到起点(根)或者是走过的点已经超过了n个,表示没有有向环
break;
}
}
if(i>n)
{
for(int i=1; i<=n; i++)if(i!=u && !flag[i]) ans+=w[pre[i]][i];
return ans;
}
//有环,进行收缩,把整个环都收缩到一个点i上。
int j=i;
memset(vis, 0, sizeof(vis));
do
{
ans += w[pre[j]][j], j=pre[j], vis[j]=flag[j]=true;//对环内的点标记,并且直接对环的权值进行加和记录,在最后找到最小树形图之后就不用展开收缩点了
}
while(j!=i);
flag[i] = false; // 环缩成了点i,点i仍然存在
//收缩点的同时,对边权值进行改变
for(int k=1; k<=n; ++k)if(vis[k]) // 在环中点点
{
for(int j=1; j<=n; j++)if(!vis[j]) // 不在环中的点
{
if(w[i][j] > w[k][j]) w[i][j] = w[k][j];
if(w[j][k]<INF && w[j][k]-w[pre[k]][k] < w[j][i])
w[j][i] = w[j][k] - w[pre[k]][k];
}
}
}
return ans;
}