树链剖分

最新推荐文章于 2024-03-24 17:40:24 发布

All_fade_away

最新推荐文章于 2024-03-24 17:40:24 发布

阅读量258

点赞数

分类专栏：数据结构-树链剖分文章标签：数据结构-树链剖分

本文链接：https://blog.csdn.net/xmr_pursue_dreams/article/details/80516980

版权

数据结构-树链剖分专栏收录该内容

1 篇文章 0 订阅

订阅专栏

树链剖分

1.何为树链剖分？

树链剖分能够通过轻重边将划分树上的链划分为轻链与重链，将树上路径操作转化为对每一个有影响的链进行操作，从而实现树上路径操作转化为多次区间操作。

2.走入树链剖分

树链剖分基于轻重边对树的划分。

下面给出一些定义：

size[x]表示以x为根的子树的节点个数。

重儿子，指一个节点的所有儿子中size最大的节点；而其他儿子为轻儿子。

重边，指一个节点连接重儿子的边；而连接其他儿子的边为轻边。

重链，指一条由重边组成的链；而其他链为轻链。

2.1树链剖分的性质

从根到任意一个树上结点的路径上最多有O（lgn）条轻链。

证明方法很多。以下是我的证明：

建模：

1.若走向轻链，size'=size-k （k>=(size/2)），代价为1。

2.若走向重链，size'=size-k （k<=(size/2)），代价为0。

由此可知：从根走到叶子节点的轻重链最多有O（lgn）条。

2.2树链剖分的实现

对于每个节点，我们需要记录：

father：父亲节点

heavy_son：重儿子

size：子树节点个数

dep：深度

top：所处重链的最上端的节点编号；若处于轻链，top为自身。

value：节点的值

d：初始时间戳

f：结束时间戳

vector<int> e：出边

需要两次dfs。

第一次dfs求出father,heavy_son,size,dep。

void dfs1(int x,int father)
{
    tree[x].father=father;
    tree[x].heavy_son=0;
    tree[x].size=1;
    tree[x].dep=tree[father].dep+1;
    for (int i=0;i<tree[x].e.size();i++)
    if (tree[x].e[i]!=father)
    {
        int to=tree[x].e[i];
        dfs1(to,x);
        tree[x].size+=tree[to].size;
        if (tree[to].size>tree[tree[x].heavy_son].size)
            tree[x].heavy_son=to; 
    }
}

第二次求出top,d,f。

void dfs2(int x,int tops)
{
    dfn++;
    tree[x].d=dfn;
    tree[x].top=tops;
    if (tree[x].heavy_son) dfs2(tree[x].heavy_son,tops);
    for (int i=0;i<tree[x].e.size();i++)
    if (tree[x].e[i]!=tree[x].father&&tree[x].e[i]!=tree[x].heavy_son) 
        dfs2(tree[x].e[i],tree[x].e[i]);
    tree[x].f=dfn;
}

即可完成树链剖分。

之后用数据结构维护重链信息进行操作即可。

3.例题：[NOI2015]软件包管理器

题目描述

Linux用户和OSX用户一定对软件包管理器不会陌生。通过软件包管理器，你可以通过一行命令安装某一个软件包，然后软件包管理器会帮助你从软件源下载软件包，同时自动解决所有的依赖（即下载安装这个软件包的安装所依赖的其它软件包），完成所有的配置。Debian/Ubuntu使用的apt-get，Fedora/CentOS使用的yum，以及OSX下可用的homebrew都是优秀的软件包管理器。

你决定设计你自己的软件包管理器。不可避免地，你要解决软件包之间的依赖问题。如果软件包A依赖软件包B，那么安装软件包A以前，必须先安装软件包B。同时，如果想要卸载软件包B，则必须卸载软件包A。现在你已经获得了所有的软件包之间的依赖关系。而且，由于你之前的工作，除0号软件包以外，在你的管理器当中的软件包都会依赖一个且仅一个软件包，而0号软件包不依赖任何一个软件包。依赖关系不存在环（若有m(m≥2)个软件包A1,A2,A3,⋯,Am，其中A1依赖A2，A2依赖A3，A3依赖A4，……，A[m-1]依赖Am，而Am依赖A1，则称这m个软件包的依赖关系构成环），当然也不会有一个软件包依赖自己。

现在你要为你的软件包管理器写一个依赖解决程序。根据反馈，用户希望在安装和卸载某个软件包时，快速地知道这个操作实际上会改变多少个软件包的安装状态（即安装操作会安装多少个未安装的软件包，或卸载操作会卸载多少个已安装的软件包），你的任务就是实现这个部分。注意，安装一个已安装的软件包，或卸载一个未安装的软件包，都不会改变任何软件包的安装状态，即在此情况下，改变安装状态的软件包数为0。

输入输出格式

输入格式：

从文件manager.in中读入数据。

输入文件的第1行包含1个整数n，表示软件包的总数。软件包从0开始编号。

随后一行包含n−1个整数，相邻整数之间用单个空格隔开，分别表示1,2,3,⋯,n−2,n−1号软件包依赖的软件包的编号。

接下来一行包含1个整数q，表示询问的总数。之后q行，每行1个询问。询问分为两种：

install x：表示安装软件包x

uninstall x：表示卸载软件包x

你需要维护每个软件包的安装状态，一开始所有的软件包都处于未安装状态。

对于每个操作，你需要输出这步操作会改变多少个软件包的安装状态，随后应用这个操作（即改变你维护的安装状态）。

输出格式：

输出到文件manager.out中。

输出文件包括q行。

输出文件的第i行输出1个整数，为第i步操作中改变安装状态的软件包数。

输入输出样例

输入样例#1：

7
0 0 0 1 1 5
5
install 5
install 6
uninstall 1
install 4
uninstall 0

输出样例#1：

输入样例#2：

10
0 1 2 1 3 0 0 3 2
10
install 0
install 3
uninstall 2
install 7
install 5
install 9
uninstall 9
install 4
install 1
install 9

输出样例#2：

说明

【样例说明 1】

一开始所有的软件包都处于未安装状态。

安装5号软件包，需要安装0,1,5三个软件包。

之后安装6号软件包，只需要安装6号软件包。此时安装了0,1,5,6四个软件包。

卸载1号软件包需要卸载1,5,6三个软件包。此时只有0号软件包还处于安装状态。

之后安装4号软件包，需要安装1,4两个软件包。此时0,1,4处在安装状态。最后，卸载0号软件包会卸载所有的软件包。`

【数据范围】

【时限1s，内存512M】

solution

树链剖分之后，用线段树维护树上信息即可。

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100005;
int dfn=0;
struct Node
{
    int top,d,f,size,dep,father,heavy_son;
    vector<int> e;
} tree[MAXN];
inline int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void dfs1(int x,int father)
{
    tree[x].size=1;
    tree[x].heavy_son=0;
    tree[x].dep=tree[tree[x].father].dep+1;
    for (int i=0;i<tree[x].e.size();i++)
    {
        dfs1(tree[x].e[i],x);
        tree[x].size+=tree[tree[x].e[i]].size;
        if (tree[tree[x].e[i]].size>tree[tree[x].heavy_son].size)
            tree[x].heavy_son=tree[x].e[i];
    }
}
void dfs2(int x,int tops)
{
    dfn++;
    tree[x].top=tops;
    tree[x].d=dfn; 
    if (tree[x].heavy_son) dfs2(tree[x].heavy_son,tops);
    for (int i=0;i<tree[x].e.size();i++)
    if (tree[x].e[i]!=tree[x].heavy_son)
        dfs2(tree[x].e[i],tree[x].e[i]);
    tree[x].f=dfn;   
}


struct Segment_tree
{
    struct NODE{int sum,value,tag,l,r;} q[MAXN<<2];
    void build(int x,int l,int r)
    {
        if ((q[x].l=l)==(q[x].r=r)) return;
        int mid=(l+r)>>1;
        build(x<<1,l,mid);
        build(x<<1|1,mid+1,r);
    }
    void down(int x,int t)
    {
        q[x].tag=1;
        q[x].value=t;
        q[x].sum=(q[x].r-q[x].l+1)*t;
    }
    void push_down(int x,int y)
    {
        q[x].tag=0;
        down(x<<1,q[x].value);
        down(x<<1|1,q[x].value);
    }
    int query(int x,int l,int r,int t)
    {
        if ((q[x].l>=l)&&(q[x].r<=r)) 
        {
            int p=q[x].sum;
            down(x,t);
            return p;
        }
        if (q[x].tag) push_down(x,t);
        int mid=(q[x].l+q[x].r)>>1;int p;
        if (r<=mid) p=query(x<<1,l,r,t);
        else if (l>mid) p=query(x<<1|1,l,r,t);
        else p=query(x<<1,l,mid,t)+query(x<<1|1,mid+1,r,t);
        q[x].sum=q[x<<1].sum+q[x<<1|1].sum;
        return p; 
    }
    int change1(int x)
    {
        int ans=0;
        for (;x;x=tree[tree[x].top].father)
            ans+=tree[x].dep-tree[tree[x].top].dep+1-query(1,tree[tree[x].top].d,tree[x].d,1);
        return ans;
    }
    int change2(int x) {return query(1,tree[x].d,tree[x].f,0); }
} segment_tree;
int main()
{
    /*
    freopen("input.in","r",stdin);
    freopen("output.out","w",stdout);
    */ 
    int n=read();
    for (int i=1;i<n;i++) 
    {
        tree[i+1].father=read()+1;
        tree[tree[i+1].father].e.push_back(i+1);
    } 
    dfs1(1,0);
    dfs2(1,1);
    segment_tree.build(1,1,n);
    int m=read();
    for (int i=1;i<=m;i++)
    {
        char st[20];
        scanf("%s",st); int x=read()+1;
        if (st[0]=='i') printf("%d\n",segment_tree.change1(x));
        else printf("%d\n",segment_tree.change2(x));
    }
    return 0;
}

4.一些总结

树链剖分是一种优秀的离线树上信息维护的算法，通过将信息的特殊归纳整理实现高效的信息处理，也是LCT的前置技能，在信息学竞赛中用途广泛（至少不用写树套树了QAQ）。

各大OJ都有不少树链剖分的题（以下题目摘自luogu）：

All_fade_away

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
树链剖分

树链剖分1.何为树链剖分？树链剖分能够通过轻重边将划分树上的链划分为轻链与重链，将树上路径操作转化为对每一个有影响的链进行操作，从而实现树上路径操作转化为多次区间操作。2.走入树链剖分树链剖分基于轻重边对树的划分。下面给出一些定义：size[x]表示以x为根的子树的节点个数。重儿子，指一个节点的所有儿子中size最大的节点；而其他儿子为轻儿子。重边，指一个节点连接重...
复制链接

扫一扫