树链剖分(轻重链剖分)
仔细想想 自己第一次听说这个这个数据结构大概有两年半的时间了 然而一直不会.
不过现在再回头来看 发现其实也不是很麻烦
首先 在学树链剖分之前最好先把 树形 以及 序这三个知识点学了
如果这三个知识点没掌握好的话 树链剖分难以理解也是当然的
-------------------------------------------------------------------------------------------
树链剖分通常用于处理树的形态不变 但点权/边权需要修改查询的题目
在选定一个点作为根后 我们来对这棵树进行操作
第一步
从根开始进行一遍遍历 (通常用 这一遍遍历需要统计以下信息
父亲节点 当前节点深度 子树大小 这些显然是在学习前置知识时已经用得比较熟练了的
然后 在这一遍遍历中 我们需要再求当前节点的重儿子
重儿子的定义是 当前节点的子节点中 子树大小最大的那个 (如果有多个任取一个 )
其余的就是轻儿子了
另外所有节点与其重/轻儿子的连边称为重/轻边
把连续的重边定义为重链
我们会发现这样一个性质
从任一节点向根节点走 走过的重链和轻边的条数都是 级别以内的
证明如下:
由于任一轻儿子对应的子树大小要小于父节点所对应子树大小的一半
因此从一个轻儿子沿轻边向上走到父节点后 所对应的子树大小至少变为两倍以上
经过的轻边条数自然是不超过 的
然后由于重链都是间断的 (连续的可以合成一条 )
所以经过的重链的条数是不超过轻边条数 的
因此经过重链的条数也是 级别的
综合可知原命题得证
从轻边向上走显然每条轻边都可以做到 向上走
而从重链向上走要做到每条重链只用 就必须额外做一些处理
第二步
利用第一遍遍历得到的信息 我们再进行一遍遍历
(需用
对于每个重儿子 要求出沿重链向上走走到顶端的点的位置
这个 显然是和父节点的一样的
对于每个轻儿子 由于向上走的重链不存在 我们可以令 为自身
现在从重链向上走都只需要 了
不过修改的复杂度仍然不靠谱
对于两点之间的修改和询问操作 轻边我们可以暴力直接修改询问 重链则必须结合一些数据结构进行以保证效率
这个时候 我们或许会回想起学 序的时候 我们将树上的点根据dfs序映射到了一维数组上
从而可以利用线段树等数据结构对在 序上连续的一段区间进行修改和询问
因此 为了能够用线段树等数据结构进行维护 我们必须将同一条重链上的点映射到一个连续的区间
这个操作并不复杂 我们只需在对每个点 时先遍历到它的重儿子 最后重链上的点映射到一维数组里便是连续的
做完这两个步骤后 树链剖分的核心部分就结束了
不过注意两个点向上走的时候 不能走得超过这两个点的
这里的具体操作最好独立思考一下 对比倍增 的写法会更好理解
看到这里 大家大概也能反应到树链剖分的复杂度是
前面的是两次dfs,然后利用线段树维护,复杂度,后面一部分是Q次查询,两个点到LCA的长度不超多logN。
以上摘自http://www.cnblogs.com/sagitta/p/5660749.html
题目大意:给你一棵树,有两种操作。第一种查询任意两点间的路径距离,第二种修改两点间的距离。
解题思路:http://www.cnblogs.com/jinkun113/p/4683299.html从这篇博客大家可以详细了解如何去用熟练剖分去解该题
#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=100010;
int fa[maxn],siz[maxn],son[maxn],w[maxn],top[maxn],dep[maxn],fp[maxn],head[maxn];
///fa为父节点,siz为子节点中siz最大的,dep为深度,son为重儿子,w表示在线段树中的位置,top为重链最顶端元素
int sum[maxn<<2];
int tree_id,n,con=0;
struct node{
int to,next;
}EE[maxn*10];
void add_edge(int u,int v){
///head[u]:u节点内第一条边序号
///EE[con].to:con边的节点
///EE[con].next:con边的下一条边序号
EE[con].to=v;EE[con].next=head[u];head[u]=con++;
}
void dfs1(int u,int ff,int deep){
son[u]=0;fa[u]=ff;siz[u]=1;dep[u]=deep;
for(int i=head[u];i!=-1;i=EE[i].next){
int v=EE[i].to;
if(v==ff) continue;
dfs1(v,u,deep+1);
siz[u]+=siz[v];
if(siz[v]>siz[son[u]]) son[u]=v;
}
}
void dfs2(int u,int ff){
w[u]=++tree_id;top[u]=ff; ///为链顶端
if(son[u]) dfs2(son[u],ff); ///遍历重儿子
else return ;///没有重儿子就内有儿子,直接返回
for(int i=head[u];i!=-1;i=EE[i].next){ ///遍历轻儿子
int v=EE[i].to;
if(v!=fa[u]&&v!=son[u]) dfs2(v,v);
}
}
void PushUP(int rt) {
sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}
void build(int l,int r,int rt) {
if (l == r) {
sum[rt] = 0;
return ;
}
int m = (l + r) >> 1;
build(lson);
build(rson);
PushUP(rt);
}
void update(int p,int add,int l,int r,int rt) {
if (l == r) {
sum[rt] = add;
return ;
}
int m = (l + r) >> 1;
if (p <= m) update(p , add , lson);
else update(p , add , rson);
PushUP(rt);
}
int query(int L,int R,int l,int r,int rt) {
if (L <= l && r <= R) {
return sum[rt];
}
int m = (l + r) >> 1;
int ret = 0;
if (L <= m) ret += query(L , R , lson);
if (R > m) ret += query(L , R , rson);
return ret;
}
int get(int u,int v){
int f1=top[u],f2=top[v],tmp=0;
while(f1!=f2){
if(dep[f1]<dep[f2]){
swap(f1,f2);
swap(u,v);
}
tmp+=query(w[f1],w[u],1,n,1);
u=fa[f1];f1=top[u];
}
if(u==v) return tmp;
if(dep[u]>dep[v]) swap(u,v);
return tmp+=query(w[son[u]],w[v],1,n,1);
}
///该题以上部分均为模板,一下部分需要手写
int U[maxn],V[maxn],C[maxn];
int main(){
int u,v,q,s,str;
while(scanf("%d%d%d",&n,&q,&s)!=-1){
memset(head,-1,sizeof(head));
memset(son,0,sizeof(son));tree_id=0;con=0;
for(int i=0;i<n-1;i++){
scanf("%d%d%d",&U[i],&V[i],&C[i]);
add_edge(U[i],V[i]);add_edge(V[i],U[i]);
}
dfs1(1,1,0);
dfs2(1,1);
memset(sum,0,sizeof(sum));
for(int i=0;i<n-1;i++){
if(dep[U[i]]>dep[V[i]]) swap(U[i],V[i]); ///利用深度低的点建边
update(w[V[i]],C[i],1,n,1);
}
while(q--){
scanf("%d",&str);
if(str==1) scanf("%d%d",&u,&v);
else scanf("%d",&u);
if(str==1) update(w[V[u-1]],v,1,n,1);
else{
printf("%d\n",get(s,u));
s=u;
}
}
}
return 0;
}