树链剖分基础（老师文档）

树链剖分

树链剖分用一句话概括就是：把一棵树剖分为若干条链，然后利用数据结构(树状数组，SBT，Splay，线段树等等)去维护每一条链，复杂度为O(logn)。

 

树链剖分的第一步是对树进行轻重边的划分：

定义size(x)为以x为根的子树节点个数，令v为u的儿子中size值最大的节点，那么(u,v)就是重边，其余边为轻边。

 

当然，关于这个它有两个重要的性质：

(1)   轻边(u,v)中，size(v)<=size(u/2)

(2)从根到某一点的路径上，不超过logn条轻边和不超过logn条重边。

 

两次dfs，第一次dfs是为了找重边，也就是记录下所有的重边。

然后第二次dfs就是连接重边形成重链，具体过程就是：以根节点为起点，沿着重边向下拓展，拉成重链，不在当前重链上的节点，都以该节点为起点向下重新拉一条重链。

 

剖分完毕后，每条重链相当于一段区间，然后用数据结构去维护，把所有重链首尾相接，放到数据结构上，然后维护整体。

 

介绍些数组，并说明各自的作用：

 

sz[]数组，用来保存以x为根的子树节点个数

 

top[]数组，用来保存当前节点的所在链的顶端节点

 

son[]数组，用来保存重儿子

 

dep[]数组，用来保存当前节点的深度

 

fa[]数组，用来保存当前节点的父亲

 

id[]数组，用来保存树中每个节点剖分后的新编号

 

Rank[]数组，用来保存当前节点在线段树中的位置

 

 

 

 

 

 

 

代码：

第一次dfs：记录所有重边

void dfs1(int u,int father,int d){

   dep[u]=d;

   fa[u]=father;

   sz[u]=1;

   for(int i=head[u];~i;i=Next[i]){

       int v=to[i];

       if(v!=father){

           dfs1(v,u,d+1);

           sz[u]+=sz[v];

           if(son[u]==-1||sz[v]>sz[son[u]]) son[u]=v;

       }

    }

}

 

第二次dfs：连重边成重链

void dfs2(int u,int tp){

   top[u]=tp;

   id[u]=++tot;

   Rank[id[u]]=u;

   if(son[u]==-1) return;//不存在重儿子时

   dfs2(son[u],tp);

   for(int i=head[u];~i;i=Next[i]){

       int v=to[i];

       //遍历u的所有儿子,如果该儿子不是重儿子,并且没有访问过。

//则以该节点为top节点开始拉一条重链。

       if(v!=son[u]&&v!=fa[u]) dfs2(v,v);

    }

}

 

 

 

 

 

 

 

 

 

 

 

 

 

 

当然，这里面有一个重要的操作，那就是修改树中边权的值。

如何修改u到v的边权的值呢？这里有两种情况：

(1)如果u与v在同一条重链上，那么就直接修改了

(2)如果u与v不在同一条重链上，那么就一边进行修改，一边将u与v往同一条重链上靠，这样就变成了第一种情况了

 

代码：

void change(int x,int y,int c){

   while(top[x]!=top[y]){

       if(dep[top[x]]<dep[top[y]]) swap(x,y);

       update(id[top[x]],id[x],c,1);

       x=fa[top[x]];

    }

   if(dep[x]>dep[y]) swap(x,y);

   update(id[x],id[y],c,1);

}