当前位置:   article > 正文

树上倍增的写法和应用(详细讲解,新手秒懂)

树上倍增


        最近做了一些树上的练习题,发现倍增真的是一种处理树上问题的神奇、方便的方法。
我以前一直打树链剖分打得多,但是学了倍增之后就再也不想打树链剖分了(当然有些题目不得不打)。
倍增比起树链剖分,代码短,容易查错,时空复杂度也优很多(nlogn),只是功能有些欠缺。


倍增的思想是二进制。

        首先开一个n×logn的数组,比如fa[n][logn],其中fa[i][j]表示i节点的第2^j个父亲是谁

然后,我们会发现有这么一个性质:

                       fa[i][j]=fa[fa[i][j-1]][j-1]

        用文字叙述为:i的第2^j个父亲 是i的第2^(j-1)个父亲的第2^(j-1)个父亲

        这是不是很神奇?这样,本来我们求i的第k个父亲的复杂度是O(k),现在复杂度变成了O(logk)。

        我们知道,一个数的二进制形式中,如果右边数第i位上是1,表示这个数如果分解为若干个2的次幂的和的形式,其中有一项一定是2^(i-1)。举个例子:10的二进制表示为1010,它的第2位和第4位上是1,所以10=2^1+2^3。

        下面是求i的第k个父亲的代码段:

  1. int father(int i,int k)
  2. {
  3. for(int x=0;x<=int(log2(k));x++)
  4. if((1<<x)&k) //(1<<x)&k可以判断k的二进制表示中,第(x-1)位上是否为1
  5. i=fa[i][x]; //把i往上提
  6. return i;
  7. }
        这样讲应该很容易理解吧?

        我们可以通过一次dfs处理出fa数组:(dep[i]表示i的深度,这个可以一起处理出来,以后要用)

如果待处理的树有n个节点,那么最多有一个节点会有2^(logn)个父亲,所以我们的fa数组第二维开logn就够了。

这里用max0表示logn。初始化fa为0,若fa[i][j]=0表示i没有第2^j个父亲。

  1. void dfs(int x)
  2. {
  3. for(int i=1;i<=max0;i++)
  4. if(fa[x][i-1]) //在dfs(x)之前,x的父辈们的fa数组都已经计算完毕,所以可以用来计算x
  5. fa[x][i]=fa[fa[x][i-1]][i-1];
  6. else break; //如果x已经没有第2^(i-1)个父亲了,那么也不会有更远的父亲,直接break
  7. for(/*每一个与x相连的节点i*/)
  8. if(i!=fa[x][0]) //如果i不是x的父亲就是x的儿子
  9. {
  10. fa[i][0]=x; //记录儿子的第一个父亲是x
  11. dep[i]=dep[x]+1; //处理深度
  12. dfs(i);
  13. }
  14. }
         这样,我们在nlogn的时间内可以通过一遍dfs处理出这棵树的相关信息。然后就可以在logn的时间内完成一些操作。

        倍增的应用中,最基础的应该就是求LCA(最近公共祖先),时间复杂度是logn。

        对于求u、v的LCA,我们可以先把u、v用倍增法把深度大的提到和另一个深度相同。如果此时u、v已经相等了,表示原来u、v就在一条树链上,直接返回此时的结果。

        如果此时u、v深度相同但不等,则证明他们的lca在更“浅”的地方,此时需要把u、v一起用倍增法上提到他们的父亲相等。为啥是提到父亲相等呢?因为倍增法是一次上提很多,所以有可能提“过”了,如果是判断他们本身作为循环终止条件,就无法判断是否提得过多了,所以要判断他们父亲是否相等。不懂的可以详见代码:

  1. int LCA(int u,int v)
  2. {
  3. if(dep[u]<dep[v])swap(u,v); //我们默认u的深度一开始大于v,那么如果u的深度小就交换u和v
  4. int delta=dep[u]-dep[v]; //计算深度差
  5. for(int x=0;x<=max0;x++) //此循环用于提到深度相同。
  6. if((1<<x)&delta)
  7. u=fa[u][x];
  8. if(u==v)return u;
  9. for(int x=max0;x>=0;x--) //注意!此处循环必须是从大到小!因为我们应该越提越“精确”,
  10. if(fa[u][x]!=fa[v][x]) //如果从小到大的话就有可能无法提到正确位置,自己可以多想一下
  11. {
  12. u=fa[u][x];
  13. v=fa[v][x];
  14. }
  15. return fa[u][0]; //此时u、v的第一个父亲就是LCA。
  16. }
        

        倍增还可以有很多变化,这让倍增法可以优更多的变化。比如用data[i][j]记录i到他的第2^j个父亲的路径长度,就可以边求LCA边求出两点距离,因为data[i][j]满足倍增的递推式:data[i][j]=data[i][j-1]+data[fa[i][j-1]][j-1]。或者用maxlen[i][j]记录i到第2^j个父亲的路径上最长边的边权,它满足maxlen[i][j]=max{maxlen[i][j-1],maxlen[fa[i][j-1]][j-1]},这样就可以快速求出两点路径上最长边的边权……

        总之,倍增是一种较为基础的处理树的方式,一定要熟练掌握,可以打几个模板题先做做,然后找一些经典的倍增习题。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/368585
推荐阅读
相关标签
  

闽ICP备14008679号