[Educational Round 3][Codeforces 609E. Minimum spanning tree for each edge]

这题本来是想放在educational round 3的题解里的,但觉得很有意思就单独拿出来写了

题目链接:609E - Minimum spanning tree for each edge

题目大意:n个点,m条边,对每条边,询问包含此边的最小生成树的边权之和

题解:大部分人都是用LCA写的,这里提供一个更为精妙的做法。

   模拟Kruskal算法建MST的过程,先将m条边按边权排序,依次进行判断。若点对(u,v)属于同一个连通块,则加入边{u,v,w}后会形成一个环,把环中最大的边换成w会多产生的权重就是包含这条边的MST比原先的MST多出来的权重,否则在MST中加入这条边。

   而这里加边的时候并不是选择直接在(u,v)之间连边,而是在他们的祖先之间连边,并有如下判断:若点v的子树大小大于点u的子树大小,则把v看做是u的父亲(将u所在的子树并入v),反之亦然。判断加边后是否会成环时,则只需暴力一层层寻找当前点的父亲即可,即是用O(n)的方法求LCA,但由于之前建树的方式用到了[small to large]的思想(就是启发式合并),所以往上爬的层数不会超过log n。因此时间复杂度为O(mlog n)。

#include<bits/stdc++.h>
using namespace std;
#define N 200001
#define LL long long
struct rua{LL u,v,w,id;}a[N];
LL n,m,mst,fa[N],f[N],sz[N],l[N];
bool cmp(rua x,rua y){return x.w<y.w;}
LL add(LL u,LL v,LL w)
{
    LL mx=0;
    while((fa[u]!=u ||fa[v]!=v) && u!=v)
      if(fa[u]==u || (fa[v]!=v && sz[v]<=sz[u]))
        mx=max(mx,l[v]),v=fa[v];
      else mx=max(mx,l[u]),u=fa[u];
    if(u==v)return w-mx;
    if(sz[u]<sz[v])swap(u,v);
    sz[u]+=sz[v],fa[v]=u,l[v]=w;
    return mst+=w,0;
}
int main()
{
    scanf("%I64d%I64d",&n,&m);
    for(int i=1;i<=m;i++)
      scanf("%I64d%I64d%I64d",&a[i].u,&a[i].v,&a[i].w),sz[i]=1,fa[i]=a[i].id=i;
    sort(a+1,a+m+1,cmp);
    for(int i=1;i<=m;i++)
      f[a[i].id]=add(a[i].u,a[i].v,a[i].w);
    for(int i=1;i<=m;i++)
      printf("%I64d\n",mst+f[i]);
}
View Code

 

posted @ 2018-09-05 15:48  DeaphetS  阅读(537)  评论(0编辑  收藏  举报