导言
你会DFS序吗?
我想,你肯定会说会.不会,欢迎点击搜索和DFS序学习
你会线段树吗?不会,欢迎点击暂无
我想,身为巨佬的你肯定会.
既然巨佬你会DFS序,会线段树.那么接下来的树链剖分,你也一定会.
接下来的学习,您必备的算法知识有,DFS序,线段树.
初学算法
适用范围
- 将树从x到y结点最短路径上所有节点的值都加上z
我们很容易发现,这个算法就是树上差分算法.
- 求树从x到y结点最短路径上所有节点的值之和
Lca大佬们,很容易发现这个其实就是Lca算法,然后通过之前我们所介绍的算法性质.
\[
dis[x]+dis[y]-dis[Lca(x,y)]
\]
就可以计算出,我们需要的答案.
但是假如说,第一问和第二问合二为一,那么我们估计就会有点懵逼了.
巨佬,时间复杂度做不到啊!
因为每一次一条路径上所有节点值都加\(z\),那么每一次都要\(DFS\)一次,重新计算\(dis\)数组.
我们发现了什么?
我们虽然只修改了一条路径上的值,但是我们却为此,去要付出了\(O(n)\)的复杂度,重新扫描一遍树,这也太不划算了.
世界上为什么要有算法?因为有奸商的存在,我们要不停地节省资源.
算法来源
一条链,一棵树,他们到底有什么关系呢?
三角关系?情敌关系?婆媳关系?后妈关系?下属关系?
其实啊,我们发现一棵树,是由很多个链构成的.
我们总能,将一棵树,拆分成几条不重叠的链.
一条链,比一棵树好处理多了.
一条链,其实可以看做成一个区间.
区间操作,是我们比较擅长的一类操作,因为他似乎有很多数据结构可以维护.
线段树,树状数组,分块,主席树,平衡树.
似乎样样都可以处理区间操作,而且复杂度都是非常低的.
而且往往,我们都会使用线段树,维护区间.
因为线段树,复杂度低,性质多多,操作多多,代码多多,烦恼多多,反正多多就对了.
假如说,我们可以把一棵树,划分成为一条条链,然后维护这些链,多好啊.
因此我们强势引出了,树链剖分算法.
算法概念
所谓的树链剖分算法,其实和他的名字一样.
就是将一棵树,划分成一些链,然后维护这些链,达到维护整棵树的作用.
某位大佬说过,算法如果可以现实化,与生活贴切,那么将会很容易理解这个算法.
我们现在考虑一个情景.
现在Acwing公司成立了,总裁y总要确立各大部门的负责人.
众所周知,不同的职业,薪水是不一样.
因此我们按照,薪水的多少,划分一下每一部门.
这些高亮的节点,就是我们每一部分,每一阶级的高薪水的大佬们.
我们发现,这些儿子节点多,也就是员工多的节点,才会高亮.
富的流油,一般都会胖的流油,而这些胖的流油,肯定也就重的流油.
所以我们把这些油光发亮的节点称之为,重儿子.
定义 | 概念 |
---|---|
重儿子 | 父亲节点的所有儿子中子树结点数目最多的结点 |
那么有富得流油,自然也就有穷的没油.逻辑黑洞,话说没油什么鬼
穷的没油,一般都会瘦的没油,而这些瘦的没油,肯定也就轻的没油.
所以我们把这些黯淡无光的节点称之为,轻儿子.
定义 | 概念 |
---|---|
轻儿子 | 父亲节点中除了重儿子以外的所有儿子节点 |
一般来说,有钱的部门,升迁都升迁的快,涨工资也涨得快,不然怎么叫作热门部门.
因此我们再来看一下这些升迁快的路径.
红色代表着升迁快的边,也就是重边.
我们发现,这些升迁快的边,都是重儿子往上走的边.
因此我们得出概念.
定义 | 概念 |
---|---|
重边 | 父亲结点和重儿子连成的边 |
有些部门升迁快,当然也有些部门升迁慢,因此我们升迁不快的边,都认为是轻边.
蓝色代表着升迁慢的边,也就是轻边.
我们发现轻边,都是轻儿子往上的边.
所以概念也就显而易见了.
定义 | 概念 |
---|---|
轻边 | 父亲节点和轻儿子连成的边 |
一时之间升迁快,不代表着有前途,一直都升迁快,才叫做前途光明
因此你想要在Acwing公司里面,找到一些前途路径.
一条红边,也可以看作前途路径.
一条前途路径,由多个升迁快的边构成,当然只有一个也木有问题.
定义 | 概念 |
---|---|
重链 | 由多条重边连接而成的路径; |
不红的路径也不少,几家欢喜几家愁,一个公司不可能天天升迁.
因此我们也会找到冷门路径.
定义快乐似神仙.
定义 | 概念 |
---|---|
轻链 | 由多条轻边连接而成的路径 |
经过上面这个故事解说下,现在秦淮岸相信各位巨佬,对于树链剖分有了一定的理解.
再次,祝愿各位巨佬们,拿到心仪的大厂offer,年薪天天往上涨,职位不停往上飞,编程水平迅速提高,脱单顺利且幸福.
概念 | 定义 |
---|---|
重儿子(节点) | 父亲节点的所有儿子中子树结点数目最多的结点; |
轻儿子(节点) | 父亲节点中除了重儿子以外的所有儿子节点 |
重边(一条边) | 父亲结点和重儿子连成的边 |
轻边(一条边) | 父亲节点和轻儿子连成的边 |
重链(多条边) | 由多条重边连接而成的路径 |
轻链(多条边) | 由多条轻边连接而成的路径 |
概念理解了,那么算法其实也就不难理解了.
其实树链剖分只是一个思想,所以我们把思想牢记于心,那么就不难.
学好算法,一定不要畏难,面对大量的知识,秦淮岸只能尽量浅显易懂地解释,所以希望大家停留在上面的表格一分钟,牢记于心.
算法梳理
我们认真审核一下树链剖分算法,理解一下树链剖分到底是要做什么,怎么处理,代码写法是什么啊,这样处理有什么好处,它的特点是什么.
树链剖分要做些什么
树链剖分,闻其名,通其意.就是把一棵树划分成几条链.
现实理解:
Acwing1班开学了,Acwing的同学们,他们互相的关系,组成一个关系树,然后我们安排座位,分成不同的小组.
提示:关系树其实一颗树,而每一个小组其实就是一条链.
怎么处理
重新编号
咱们知道,一个班级的管理,是有很多编排方式,但是最常见的方式,就是学号编排.
所以为了更好的管理一个班级,为什么不使用学号编排呢.
因此咱们使用学号编排,不过呢这个学号编排有独特之处.
小A请默认是自己.
加入现在操场上,所有人按照以上顺序站着,然后y总拿着达达牌小橙笔,开始了标记学号.
以上这张图片,非常重要,如果不懂,一定要看蒟蒻的直播讲解.广告时间到了
现在学号编辑完毕了,那么现在的重点,就是分小组了.
分类小组
咱们知道一个小组,肯定有一个核心人物红太阳.
那么谁是咱们的红太阳,红太阳肯定走在最前面,迎接着晨曦.
所以一个小组的组长红太阳,一定是一条链的链头.
对于重链而言,他的链头,一定就是深度最浅的那一位.
不然肯定有人玩着篮球踩在它的头上.
对于轻链而言,他的链头,就是他自己,毕竟是孤寡老人.
总结讨论
总结一下,我们需要哪些元素.
- 深度(为了下面服务)
- 父亲节点
- 子树大小(团队大小)
- 重儿子(身宽体胖)
- DFS序(新学号)
- 链顶(红太阳)
- 老编号(曾经的编号,其实就是名字,
毕竟A遇到B,总不能一直喊人家叫做XX号)
一个表格设置一下.
定义 | 含义 |
---|---|
\(deep[x]\) | 节点\(x\)的深度 |
\(fa[x]\) | 节点\(x\)的父亲 |
\(size[x]\) | 节点\(x\)的子树大小 |
\(wson[x]\) | 节点\(x\)的重儿子的老编号(名字). |
\(dfn[x]\) | 节点\(x\)的新编号,其实也就是DFS序 |
\(top[x]\) | 节点\(x\)所在链的链头 |
\(pre[x]\) | 节点\(x\)的老编号,也就是老名字. |
总而言之,这就是我们的目标数组了,看上去好多啊,但其实只要两次DFS就解决完毕了.
代码解析
void dfs1(int x,int father)//father是x的父亲节点
{
size[x]=1;//刚开始子树为1,也就是直接
for(int i=head[x]; i; i=Next[i])//访问所有出边
{
int y=edge[i];//儿子节点
if (y==father)//访问到唯一一个不是儿子节点的父亲节点去了
continue;//当然不可以,直接跳过
deep[y]=deep[x]+1;//深度+1,儿子节点是父亲节点深度+1
fa[y]=x;//y的父亲节点是x
dfs1(y,x);//y的父亲节点是x
size[x]+=size[y];//加上儿子贡献的子树
if (size[y]>size[wson[x]])//如果这个节点比当前重儿子,子树还要多(还要重) ,那么营养过剩的重儿子就是他了.
wson[x]=y;//wson[x]表示x节点的重儿子
}
}
这一层DFS,就让我们求解出来了前四个数组,那么后三个数组呢?
void dfs2(int x,int tp)//x表示当前节点,tp表示链头
{
dfn[x]=++cnt;//重新编辑编号
pre[cnt]=x;//存储老编号
top[x]=tp;//存储链头
if (wson[x]) //优先处理重儿子
dfs2(wson[x],tp);
for(int i=head[x]; i; i=Next[i])//访问所有的轻链
{
int y=edge[i];//存储出边
if (y==wson[x] || y==fa[x])//重儿子已经处理过了,父亲节点不可以抵达
continue;
dfs2(y,y);//每一个轻链的链头,都是自己,而且重儿子的开头都是轻链.
}
}
算法拓展
咱们知道,树链剖分是LCA+树上差分的合并增强版,那么他们两个有的操作,他当然也得有.
正如同某名言.你的是我的,我的还是我的.
先来看LCA的操作,其实就是其名的求两点的最近公共祖先.
我们对于上面所述的top数组,也就是链头数组,比较懵逼,觉得他似乎没有什么用处啊.
实际上它的用处非常之大,是树链剖分的核心数组之一.
通俗理解的故事,又来了.以下内容纯属瞎编,只为了更好的理解.
\(Acwing\)学校开课了,孩子算法学不好,怎么办?
多半是找不到好老师,快来参加\(Acwing\)算法实验班.
现在知道\(Acwing\)学校里面实力最为高深莫测就是y总,但是不是所有人都认识y总.
现在小\(A\)同学想知道自己学校的校长是谁,但是他是新来的新生.
小\(A\)同学只认识自己的组长\(song666\)巨佬.
\(song666\)巨佬只知道年级组长\(Acwing1901\)班的熊熊老师.
熊熊老师知道y总,所以咱们得出了以下这条最短路径.
\[ 小A->song666->Acwing1901班熊熊老师->y总 \]
如果说所有人都不会树链剖分算法的top链头数组,那么将会这样询问.
我们假设熊熊老师是年级组长.(原谅我只认识熊熊老师)
\[
小A->小B->小C->小D->song666巨佬-> \\\\
Acwing1905班主任媛媛老师->Acwing1904班丰丰老师->Acwing1903班辰辰老师 \\\\
->Acwing1902班彬彬老师->Acwing1901班熊熊年级组长->y校长.
\]
我们发现链头数组,其实可以形象的认为,一个小组的领头人物.
重点来了,注意力迅速集中.
假如说节点A在A小组,然后节点B在B小组,要我们找\(Lca(a,b)\).
那么我们得出.
因为A小组和B小组是两个没有交集的小组.(也就是互相都不认识)
那么显然A小组这一串人,都不可能成为答案.
同理B小组这一伙人,也不可能成为答案.
明知没有用,何必浪费时间.
所以,我们迅速跨越中间的所有节点.
a=top[a];//直接爬到链头
b=top[b];//直接爬到链头
这样我们发现效率直线上升.
\[
本来我们a节点爬到这条链需要花费O(len(a)).
\]
提示一下
\[
提示:len(a)表示这条a所在链的长度.
\]
但是现在得了链头点,一个传染两.
于是时间效率突飞猛进.
\[
O(1)
\]
这就是树链剖分的完美之处.
然后我们知道从任意一个节点,抵达根节点的链数量,不会超过\(O(logn)\)条重链,和\(O(logn)\)条轻链.
因此我们查询复杂度的极限是.
\[
O(log^2n)
\]
看上去比Lca差一些啊,倍增Lca查询的时间复杂度是\(O(logn)\)
事实上很多时候复杂度小的可怜,比如说.
假如说A和B在同一个小组,而且B是组长,那么复杂度\(O(1)\)
拓展代码
int Query_sum(int a,int b)//a,b这条路径上的权值和
{
int ans=0;//初始化为0
while(top[a]!=top[b]) //此时a,b不在同一条重链上面
{
if (deep[top[a]]<deep[top[b]])//我们这里默认a是深度在下面的链
swap(a,b);
ans+=t.Query_sum(1,1,n,dfn[top[a]],dfn[a]);//访问这条重链,dfn[top[a]]是链头,dfn[a]是当前节点,dfn[top[a]]<dfn[a]因为a后访问.
a=fa[top[a]];//往上面访问,也就是往上面爬一步
}
if (deep[a]<deep[b])//保证a深度比b深度,深.保证后面查询l<r
swap(a,b);
ans+=t.Query_sum(1,1,n,dfn[b],dfn[a]);//他们已经在同一条重链上面了
return ans;
}
链上结构
路径操作
看到上面的代码,我们发现t是什么东西.
t其实就是我们的线段树数据结构.
根据树上差分的操作们,得知我们的树链剖分需要复制一大波操作.
将树从x到y结点最短路径上所有节点的值都加上z
求树从x到y结点最短路径上所有节点的值之和
根据上面这些操作,我们不难发现,每条链也需要资瓷这些操作.
综上所述得出.
- 一条链上,需要资瓷区间修改.
- 一条链上,需要支持区间查询.
总而言之,区间操作多多,因此我们使用线段树.
当然了巨佬们,肯定喜欢使用平衡树等等高大上,上档次,有内涵的数据结构,但是线段树的代码复杂度对于考试而言,是最好不过的结构了.
代码复杂度是考试的时候,极为重要的复杂度. by Acwing站长,校长,集训队大佬yxc总裁
总而言之,言而总之,我们树链剖分的代码量,成功的增加了1k.
子树操作
我们再来几个操作.
将以x为根节点的子树内所有节点值都加上z
求以x为根节点的子树内所有节点值之和
看到子树操作,有点不知所措,线段树似乎不支持这个鬼东西吧,难道代码量又要翻倍处理.
事实上,我们的代码只需要增加八行代码,也就核心两句话.
我们的DFS遍历,在这里起到了决定性,关键性,核心性的作用.(语文老师:捕捉到了病句,起到了,什么的作用,句式杂糅)
我们发现DFS序列的一个重点,就是.一颗子树,它的DFS序,是有序的.
还是这张图,我们发现熊熊助教这颗子树.
其实就是\([2,7]\)的这个区间.
因此我们只要修改\([2,7]\)这个区间,就达到了修改以2为根的树修改.
因此代码如下.
void Update(int x,int v)//修改子树的值
{
t1.Update(1,1,n,dfn[x],dfn[x]+size[x]-1,v);//DFS序是有序的数列
}
同理,查询操作也就如下所示了.
int Query_sum(int x)//查询子树的和
{
return t1.Query_sum(1,1,n,dfn[x],dfn[x]+size[x]-1);//统计子树和,其实和修改差不多
}
树链模板
struct line_tree//线段树
{
#define mid (l+r>>1)//二分中点
#define Lson (rt<<1),l,mid//左儿子
#define Rson (rt<<1 | 1),mid+1,r//右儿子
#define Len (r-l+1)//区间长度
void Push_down(int rt,int l,int r)
//这里的懒惰标记只适合区间修改,把[l,r]区间都变成v.如果是都+v,则必须Lazy标记+=,而不是下面的=
{
if (Lazy[rt]!=-1)//当前节点有Lazy标记
{
Lazy[rt<<1]=Lazy[rt<<1 | 1]=Lazy[rt];//懒惰标记下传
sum[rt<<1]=Lazy[rt]*(mid-l+1);//左儿子的区间长度
sum[rt<<1 |1]=Lazy[rt]*(r-(mid+1)+1);//右儿子的区间长度
Lazy[rt]=-1;//此时懒惰标记已经下传完毕了,那么可以全部清空了
}
}
void build(int rt,int l,int r)
{
Lazy[rt]=-1;
if (l==r)//叶子节点
{
sum[rt]=1;//sum数组
return ;
}
build(Lson);//左儿子
build(Rson);//右儿子
sum[rt]=sum[rt<<1]+sum[rt<<1 |1];//左儿子节点+右儿子节点
}
void Update(int rt,int l,int r,int L,int R,int v)//将[L,R]区间统统修改成为v,然后当前区间[l,r]
{
if (L<=l && r<=R)//当前区间被包括了
{
Lazy[rt]=v;//懒惰标记修改
sum[rt]=v*Len;//全部修改完毕
return ;
}
Push_down(rt,l,r);//向下传递Lazy标记
if (L<=mid)//在左儿子身上
Update(Lson,L,R,v);//左儿子
if (R>mid)//在右儿子身上
Update(Rson,L,R,v);//右儿子
sum[rt]=sum[rt<<1]+sum[rt<<1 |1];//左儿子+右儿子
}
int Query_sum(int rt,int l,int r,int L,int R)//查询[L,R]区间和,当前区间[l,r]
{
int ans=0;
if (L<=l && r<=R)//当前区间被包括了
return sum[rt];
if (L<=mid)//左儿子上面
ans+=Query_sum(Lson,L,R);
if (R>mid)//右儿子上面
ans+=Query_sum(Rson,L,R);
sum[rt]=sum[rt>>1]+sum[rt>>1 |1];
return ans;//返回
}
} t1;
struct Tree_Chain//树链剖分
{
void add_edge(int a,int b)//添加边函数
{
edge[++tot]=b;//出边节点
Next[tot]=head[a];//链表链接
head[a]=tot;//上一个节点
}
void dfs1(int x,int father)//x节点,和他的父亲节点father
{
size[x]=1;//刚开始就自己这一个节点
for(int i=head[x]; i; i=Next[i]) //开始遍历所有的出边
{
int y=edge[i];//出边节点
if (y==father)//儿子节点是不可以等于父亲节点的
continue;
deep[y]=deep[x]+1;//儿子的深度,是父亲深度+1
fa[y]=x;//y的父亲节点是x
dfs1(y,x); //开始遍历儿子节点
size[x]+=size[y];//儿子节点贡献子树大小
if (size[y]>size[wson[x]]) //发现当前的儿子节点,比之前的重儿子,还要重(胖),那么更新重儿子
wson[x]=y;//更新
}
return ;//华丽结束
}
void dfs2(int x,int tp)//x节点,以及x节点所在链的链头
{
dfn[x]=++cnt;//当前节点的新编号,也就是DFS序编号
pre[cnt]=x;//老编号,虽然在整道题目中没有用处,但是树链剖分板子打一遍也是好的
top[x]=tp;//链头存储一下
if (wson[x])//有重儿子,那么一定先访问重儿子
dfs2(wson[x],tp); //访问节点,此时重儿子一定在重链上,所以还是tp
for(int i=head[x]; i; i=Next[i]) //访问所有的轻儿子
{
int y=edge[i];//出边
if (y==wson[x] || y==fa[x])//轻儿子节点不能是重儿子,也不能是父亲节点
continue;
dfs2(y,y);//每一个轻儿子,他的链头其实都是自己,而且重链的开头也得是轻儿子
}
}
void Update(int x,int v)//修改子树的值
{
t1.Update(1,1,n,dfn[x],dfn[x]+size[x]-1,v);//DFS序是有序的数列
}
int Query_sum(int x)//查询子树的和
{
return t1.Query_sum(1,1,n,dfn[x],dfn[x]+size[x]-1);//统计子树和,其实和修改差不多
}
long long Query_sum2(int a,int b)//a,b这条路径上的权值和
{
long long ans=0;//初始化为0
while(top[a]!=top[b]) //此时a,b不在同一条重链上面
{
if (deep[top[a]]<deep[top[b]])//我们这里默认a是深度在下面的链
swap(a,b);
now_ans=0;
t1.Query_sum(1,1,n,dfn[top[a]],dfn[a]);
ans+=now_ans;//访问这条重链,dfn[top[a]]是链头,dfn[a]是当前节点,dfn[top[a]]<dfn[a]因为a后访问.
a=fa[top[a]];//往上面访问,也就是往上面爬一步
}
if (deep[a]<deep[b])//保证a深度比b深度,深.保证后面查询l<r
swap(a,b);
now_ans=0;
t1.Query_sum(1,1,n,dfn[top[a]],dfn[a]);
ans+=now_ans;//他们已经在同一条重链上面了
return ans;
}
int Update2(int a,int b) //链上修改
{
while(top[a]!=top[b])//也就是两个点还在不同的重链上
{
if (deep[top[a]]<deep[top[b]])//我们默认a节点是深度比b节点深一些
swap(a,b);//交换一下就好了
t1.Update(1,1,n,dfn[top[a]],dfn[a],0);//在爬的过程中,也帮忙修改一下
a=fa[top[a]];//往上面爬一下
}
if (deep[a]<deep[b])//保证a深度比b深度,深.保证后面查询l<r
swap(a,b);
t1.Update(1,1,n,dfn[b],dfn[a],0);//同一条链了,那么修改这条链上在[a,b]之间的点
}
//请注意,本模板是很多题目的操作合并而成,可能有问题,但是单独没有问题.如果有问题请艾特博主.
//所有代码风格为博主风格,所以操作之间都是通用的,应该不会出现问题.
} t2;
经典选讲
暂无中.
标签:链剖分,剖分,int,儿子,树链,我们,节点 来源: https://www.cnblogs.com/gzh-red/p/11235038.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。