树链剖分

定义

将一棵树拆分成重边和轻边:
重边:一个节点与size(size表示子树节点个数)最大的儿子之间的边(当然叶节点没有)。
轻边;除重边之外的边。
ps:重边连接的儿子叫做重儿子,轻边连接的儿子叫做轻儿子。

一段重边连起来形成的链叫做重链。

作用

根据定义,可以得到初步结论:
size(轻儿子)<=size(父节点)/2

根据这个初步结论,可以得出推论:
n表示节点数,则从根到任何一个节点,轻边的个数<=log2(n)。

又可以得到:
从根到任何一个节点,重链的个数<=log2(n)
这个有些难推,这里说明一下:
因为重链是重边全部连起来形成的,那么重链最多的时候是重边与轻边交替出现,而轻边的个数<=log2(n),则重链的个数也<=log2(n)。

有了这些性质,再套上一些数据结构(比如线段树),我们就可以在log2(n)的时间内修改树上的一段路径了。

实现

为了使用线段树,我们先构造几个数组:
si[x]:x子树的节点个数
fa[x]:x的父节点
sonH[x]:x的重儿子
dep[x]:x的深度
top[x];x沿着重边向上走能走到的最远节点
id[x]:x的编号
who[x]:编号为x的节点

其中si,fa,sonH,dep通过普通Dfs遍历就可以实现,代码如下:

void Dfs(int x)
{
    dep[x]=dep[fa[x]]+1;si[x]=1;
    for (int j=lnk[x];j;j=nxt[j])
        if (son[j]!=fa[x])
        {
            fa[son[j]]=x;
            Dfs(son[j]);
            if (si[son[j]]>si[sonH[x]]) sonH[x]=son[j];
            si[x]+=si[son[j]];
        }
}

为了让一条重链的所有节点的编号有序,在构造top,id和who的时候优先走重儿子,代码如下:

void HLD(int x,int lst)
{
    top[x]=lst;id[x]=++tot;who[tot]=x;
    if (sonH[x]) HLD(sonH[x],lst); //优先走重儿子
    for (int j=lnk[x];j;j=nxt[j])
        if (son[j]!=fa[x]&&son[j]!=sonH[x]) HLD(son[j],son[j]);
}

然后最主要的问题就是如何快速让x和y都走到根(和LCA目的一样,所以求最近公共祖先的时候可以很愉快的用树链剖分,而且空间是线性级别的),其实想法和LCA差不多,只不过一般要套上个线段树求一些信息比如总和、最大值:
1.令fx=top[x],fy=top[y]
2.如果fx==fy(fx和fy在一条重链上或x==y)跳到第5步。
3.如果dep[fx]<dep[fy],交换x和y。
4.求id[fx]~id[x](根据上面的HLD函数,fx的编号肯定比x小)之间的信息,然后让x=fa[fx],跳回第1步。
5.如果id[x]>id[y],交换x和y。求id[x]~id[y]之间的信息。
实现实际上也很简单,伪代码如下:

int Ask(int x,int y)
{
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        求id[top[x]]~id[x]之间的信息。
        x=fa[top[x]];
    }
    if (id[x]>id[y]) swap(x,y);
    求id[x]~id[y]之间的信息。
    return 答案。
}

至此,树链剖分就讲完了。

模板

以BZOJ1036为例

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=30000,maxm=maxn,MAXINT=((1<<30)-1)*2+1;

int n,te,E,num[maxn+5];
int lnk[maxn+5],son[2*maxm+5],nxt[2*maxm+5];
int si[maxn+5],dep[maxn+5],fa[maxn+5],sonH[maxn+5];
int top[maxn+5],id[maxn+5],who[maxn+5],tot;
//=================================================
bool Eoln(char ch) {return ch==10||ch==13||ch==EOF;}
int readi(int &x) //读入优化
{
    int tot=0,f=1;char ch=getchar(),lst=ch;
    while ('9'<ch||ch<'0') {if (ch==EOF) return EOF;lst=ch;ch=getchar();}
    if (lst=='-') f=-f;
    while ('0'<=ch&&ch<='9') tot=tot*10+ch-48,ch=getchar();
    x=tot*f;
    return Eoln(ch);
}
char getrch() {char ch=getchar();while (ch!='C'&&ch!='H'&&ch!='Q'&&ch!='S'&&ch!='M') ch=getchar();return ch;}
//=================================================
void Add(int x,int y) {son[++E]=y;nxt[E]=lnk[x];lnk[x]=E;}
void Dfs(int x)
{
    dep[x]=dep[fa[x]]+1;si[x]=1;
    for (int j=lnk[x];j;j=nxt[j])
        if (son[j]!=fa[x])
        {
            fa[son[j]]=x;
            Dfs(son[j]);
            if (si[son[j]]>si[sonH[x]]) sonH[x]=son[j];
            si[x]+=si[son[j]];
        }
}
void HLD(int x,int lst)
{
    top[x]=lst;id[x]=++tot;who[tot]=x;
    if (sonH[x]) HLD(sonH[x],lst);
    for (int j=lnk[x];j;j=nxt[j])
        if (son[j]!=fa[x]&&son[j]!=sonH[x]) HLD(son[j],son[j]);
}
//=================================================
struct LT //线段树
{
    int l[4*maxn+5],r[4*maxn+5],sum[4*maxn+5],MAX[4*maxn+5];
    void Build(int id,int L,int R)
    {
        l[id]=L;r[id]=R;
        if (L==R) {sum[id]=MAX[id]=num[who[L]];return;}
        int mid=L+(R-L>>1);
        Build(id*2,L,mid);Build(id*2+1,mid+1,R);
        Pushup(id);
    }
    void Pushup(int id) {sum[id]=sum[id*2]+sum[id*2+1];MAX[id]=max(MAX[id*2],MAX[id*2+1]);}
    void Change(int id,int L,int x)
    {
        if (L<l[id]||r[id]<L) return;
        if (l[id]==r[id]) {sum[id]=x;MAX[id]=x;return;}
        Change(id*2,L,x);Change(id*2+1,L,x);
        Pushup(id);
    }
    int Ask_sum(int id,int L,int R)
    {
        if (R<l[id]||r[id]<L) return 0;
        if (L<=l[id]&&r[id]<=R) return sum[id];
        return Ask_sum(id*2,L,R)+Ask_sum(id*2+1,L,R);
    }
    int Ask_max(int id,int L,int R)
    {
        if (R<l[id]||r[id]<L) return -MAXINT;
        if (L<=l[id]&&r[id]<=R) return MAX[id];
        return max(Ask_max(id*2,L,R),Ask_max(id*2+1,L,R));
    }
};
LT tr;
//=================================================
int Ask_sum(int x,int y) //询问总和
{
    int sum=0;
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        sum+=tr.Ask_sum(1,id[top[x]],id[x]);
        x=fa[top[x]];
    }
    if (id[x]>id[y]) swap(x,y);
    sum+=tr.Ask_sum(1,id[x],id[y]);
    return sum;
}
int Ask_max(int x,int y) //询问最大值
{
    int MAX=-MAXINT;
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        MAX=max(MAX,tr.Ask_max(1,id[top[x]],id[x]));
        x=fa[top[x]];
    }
    if (id[x]>id[y]) swap(x,y);
    MAX=max(MAX,tr.Ask_max(1,id[x],id[y]));
    return MAX;
}
int main()
{
    freopen("HLD.in","r",stdin);
    freopen("HLD.out","w",stdout);
    readi(n);
    for (int i=1;i<=n-1;i++)
    {
        int x,y;readi(x);readi(y);
        Add(x,y);Add(y,x);
    }
    for (int i=1;i<=n;i++) readi(num[i]);
    Dfs(1);HLD(1,1);tr.Build(1,1,n);
    readi(te);
    while (te--)
    {
        char ch1=getrch(),ch2=getrch();
        int x,y;readi(x);readi(y);
        if (ch1=='C') tr.Change(1,id[x],y); else
        if (ch1=='Q') if (ch2=='S') printf("%d\n",Ask_sum(x,y)); else
        printf("%d\n",Ask_max(x,y));
    }
    return 0;
}
  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值