线段树合并背景
如果有若干棵线段树,它们都维护相同的值域 [ 1 , n ] [1,n] [1,n],那么它们对各个子区间的划分显然是一致的。假如有 m m m次单点修改操作,每次操作在某一棵线段树上执行,所有操作完成后,我们希望把这些线段树对应位置上的值相加,同时维护区间最大值。此时我们就可以使用线段树合并来解决。
线段树合并
我们依次合并这些线段树。合并两棵线段树时,用两个指针
p
,
q
p,q
p,q从两个根节点,以递归的方式同步遍历两棵线段树。换句话说,
p
,
q
p,q
p,q指向的节点总是代表相同的区间。
1.若
p
,
q
p,q
p,q之一为空,则以非空的那个作为合并后的节点。
2.若
p
,
q
p,q
p,q均不为空,则递归合并两棵左子树和两棵右子树,然后删除节点
q
q
q,以
p
p
p为合并后的节点,自底向上更新最值信息。若已经达到叶子节点,则直接把两个最值相加即可。
inline void pushup(int p){t[p].dat=max(t[t[p].lc].dat,t[t[p].rc].dat);}
int merge(int p,int q,int l,int r){
if(!p)return q;
if(!q)return p;
if(l==r){
t[p].dat+=t[q].dat;
return p;
}
int mid=(l+r)/2
t[p].lc=merge(t[p].lc,t[q].lc,l,mid);
t[p].rc=merge(t[p].rc,t[q].rc,mid+1,r);
pushup(p);
return p;
}
仔细观察就可以发现,若线段树合并过程中发生递归,则必定会导致 p , q p,q p,q之一被删除。因此,在完成所有线段树的合并后, m e r g e merge merge函数被执行的次数不会超过所有线段树的节点总数加 1 1 1。故整个合并过程的时间复杂度是 O ( m log n ) O(m\log n) O(mlogn),非常高效。