[note]树链剖分
树链剖分https://www.luogu.org/problemnew/show/P3384
概念
树链剖分,是一种将树剖分成多条不相交的链的算法,并通过其他的数据结构来维护这些链上的信息。
最简单的例子就是\(LCA\),假设现在有一棵退化成链的树。如果要求任意两点的\(LCA\),因为他们在同一条链上的缘故,只需要判断一下两者的深度就行了。由此可见,在链上是比在树上更好操作的。
那么该怎么将一棵树剖分开来捏?
先搬出一堆概念:
重儿子
在以X节点为根的子树中,节点数最多的子树的根节点,即是X节点的重儿子。
重边
连接X节点与X节点的重儿子的边,我们叫他重边。
重链
一堆重边连起来的链。
轻链
一堆非重边连起来的链。
对于每个节点,找出其重儿子,就可以剖分成一条条重链与轻链。
实现
数组定义
\(val[N]\)每个节点的初值
\(size[N]\)每个节点子树的大小
\(son[N]\)每个节点的重儿子
\(fa[N]\)每个节点的父亲
\(dfn[N]\)每个节点在线段树上的编号
\(rk[N]\)线段树上节点在树中的编号
\(dep[N]\)节点深度
\(top[N]\)每个点所在链的链顶
\(dfs\)
第一遍\(dfs\)处理出\(size[],fa[],son[],dep[]\)
第二遍\(dfs\)处理出\(top[],dfn[],rk[]\)
线段树
用线段树维护树链,并实现链上的操作,常见操作如下:
将树从\(X\)到\(Y\)结点最短路径上所有节点的值都加上\(Z\)
求树从\(X\)到\(Y\)结点最短路径上所有节点的值之和
将以\(X\)为根节点的子树内所有节点值都加上\(Z\)
求以\(X\)为根节点的子树内所有节点值之和
对于子树操作:我们知道一颗子树内的编号一定是连续的,那么以\(X\)节点为根的子树的区间就是\((dfn[x],dfn[x]+size[x]-1)\)
#define RG register
#include<cstdio>
#include<iostream>
using namespace std;
const int N=1e5+5;
inline int read()
{
RG int x=0,w=1;RG char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
return x*w;
}
int n,m,r,p,cnt,ct;
int lazy[N<<2],sum[N<<2];
int val[N],size[N],son[N],fa[N],dfn[N],rk[N],last[N],dep[N],top[N];
struct edge{
int to,next,w;
}e[N<<1];
void insert(int u,int v)
{
e[++cnt]=(edge){v,last[u]};last[u]=cnt;
e[++cnt]=(edge){u,last[v]};last[v]=cnt;
}
inline void dfs1(int now)
{
size[now]=1;
for(int i=last[now];i;i=e[i].next)
{
int v=e[i].to;
if(v==fa[now])continue;
fa[v]=now;
dep[v]=dep[now]+1;
dfs1(v);
size[now]+=size[v];
if(size[v]>size[son[now]])son[now]=v;
}
}
inline void dfs2(int now,int Top)
{
top[now]=Top;dfn[now]=++ct;rk[ct]=now;
if(son[now])dfs2(son[now],Top);//重儿子
for(int i=last[now];i;i=e[i].next)
{
int v=e[i].to;
if(v==son[now]||v==fa[now])continue;
dfs2(v,v);//轻儿子的top就是本身
}
}
inline void Pushup(int root){sum[root]=(sum[root<<1]+sum[root<<1|1])%p;}
void Build(int root,int l,int r)//建树
{
if(l==r){sum[root]=val[rk[l]];return;}
int mid=(l+r)>>1;
Build(root<<1,l,mid);
Build(root<<1|1,mid+1,r);
Pushup(root);
}
inline void Pushdown(int root,int len)//下放懒标记,len为区间长度
{
if(!lazy[root])return;
lazy[root<<1]=(lazy[root<<1]+lazy[root])%p;
lazy[root<<1|1]=(lazy[root<<1|1]+lazy[root])%p;
sum[root<<1]=(sum[root<<1]+lazy[root]*(len-(len>>1))%p)%p;
sum[root<<1|1]=(sum[root<<1|1]+lazy[root]*(len>>1)%p)%p;
lazy[root]=0;
}
void Modify(int root,int l,int r,int ll,int rr,int k)//区间修改
{
Pushdown(root,r-l+1);
if(ll<=l&&r<=rr)
{
lazy[root]=(lazy[root]+k)%p;
sum[root]=(sum[root]+k*(r-l+1)%p)%p;
return;
}
int mid=(l+r)>>1;
if(ll<=mid)Modify(root<<1,l,mid,ll,rr,k);
if(rr>mid)Modify(root<<1|1,mid+1,r,ll,rr,k);
Pushup(root);
}
int Query(int root,int l,int r,int ll,int rr)//区间查询
{
Pushdown(root,r-l+1);
if(ll<=l&&r<=rr)return sum[root];
int mid=(l+r)>>1,Sum=0;
if(ll<=mid)Sum=(Sum+Query(root<<1,l,mid,ll,rr))%p;
if(mid<rr)Sum=(Sum+Query(root<<1|1,mid+1,r,ll,rr))%p;
Pushup(root);
return Sum;
}
inline void Modify_Tree(int x,int y,int k)//修改节点x到节点y路径上点的点权
{
while(top[x]!=top[y])//不在同一条链上
{
if(dep[top[x]]<dep[top[y]])swap(x,y);//注意是比较链顶的深度
Modify(1,1,n,dfn[top[x]],dfn[x],k);//链顶更深的节点跳链
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
Modify(1,1,n,dfn[x],dfn[y],k);//同一条链上,直接区间查询
}
inline int Query_Tree(int x,int y)//查询节点x到节点y路径上点权和
{
int Sum=0;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
Sum=(Sum+Query(1,1,n,dfn[top[x]],dfn[x]))%p;
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
return (Sum+Query(1,1,n,dfn[x],dfn[y]))%p;
}
inline void Modify_Son(int x,int k)//子树修改
{
Modify(1,1,n,dfn[x],dfn[x]+size[x]-1,k);
}
inline int Query_Son(int x)
{
return Query(1,1,n,dfn[x],dfn[x]+size[x]-1);//子树查询
}
int main()
{
n=read(),m=read(),r=read(),p=read();
for(int i=1;i<=n;i++)val[i]=read()%p;
for(int i=1;i<n;i++)insert(read(),read());
dep[r]=1;
dfs1(r);
dfs2(r,r);
Build(1,1,n);
while(m--)
{
int f=read();
if(f==1){int x=read(),y=read(),z=read()%p;Modify_Tree(x,y,z);}
if(f==2)printf("%d\n",Query_Tree(read(),read()));
if(f==3){int x=read(),y=read()%p;Modify_Son(x,y);}
if(f==4)printf("%d\n",Query_Son(read()));
}
return 0;
}
[note]树链剖分的更多相关文章
- CF 191C Fools and Roads lca 或者 树链剖分
They say that Berland has exactly two problems, fools and roads. Besides, Berland has n cities, popu ...
- CodeForces 916E Jamie and Tree(树链剖分+LCA)
To your surprise, Jamie is the final boss! Ehehehe. Jamie has given you a tree with n vertices, numb ...
- Codeforces Round #425 (Div. 2) Problem D Misha, Grisha and Underground (Codeforces 832D) - 树链剖分 - 树状数组
Misha and Grisha are funny boys, so they like to use new underground. The underground has n stations ...
- Housewife Wind(边权树链剖分)
Housewife Wind http://poj.org/problem?id=2763 Time Limit: 4000MS Memory Limit: 65536K Total Submis ...
- HDU3710 Battle over Cities(最小生成树+树链剖分+倍增+线段树)
Battle over Cities Time Limit: 10000/5000 MS (Java/Others) Memory Limit: 65536/65536 K (Java/Othe ...
- D. Happy Tree Party CodeForces 593D【树链剖分,树边权转点权】
Codeforces Round #329 (Div. 2) D. Happy Tree Party time limit per test 3 seconds memory limit per te ...
- BZOJ3531 SDOI2014 旅行 - 树链剖分,主席树
题意:给定一棵树,树上每个点有权值和类型.支持:修改某个点的类型:修改某个点的权值:询问某条链上某个类型的点的和/最大值.点数/类型数/询问数<=100000. 分析: 树链剖分,对每个类型的点 ...
- BZOJ 3626: [LNOI2014]LCA [树链剖分 离线|主席树]
3626: [LNOI2014]LCA Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 2050 Solved: 817[Submit][Status ...
- BZOJ 1984: 月下“毛景树” [树链剖分 边权]
1984: 月下“毛景树” Time Limit: 20 Sec Memory Limit: 64 MBSubmit: 1728 Solved: 531[Submit][Status][Discu ...
随机推荐
- 转:Hash, MAC,HMAC说明
from: http://www.cnblogs.com/songhan/archive/2012/07/29/2613898.html Hash, MAC,HMAC Hash-MD5, SHA-1, ...
- 联想Y430P CentOS 7.3 无线网络的配置
# uname -a # 查看内核/操作系统/CPU信息的Linux系统信息命令 [root@www ~]# uname -a Linux www SMP Tue Nov :: UTC x86_64 ...
- 12. 改变JDK编译版本【从零开始学Spring Boot】
spring Boot在编译的时候,是有默认JDK版本的,如果我们期望使用我们要的JDK版本的话,那么要怎么配置呢? 这个只需要修改pom.xml文件的<build>-- <plug ...
- RFS 理解
1.背景 网卡接收一个数据包的情况下,会经过三个阶段: - 网卡产生硬件中断通知CPU有包到达 - 通过软中断处理此数据包 - 在用户态程序处理此数据包 在SMP体系下,这三个阶段有可能在3个 ...
- python实现大文件分割与合并
小U盘传大电影时可以免去用winrar分割文件时的压缩和解压缩过程. file.py import sys from os.path import exists fileCount = 0 def s ...
- spring拦截器不拦截方法名原因
开发一个基于注解的登录拦截器,遇到拦截器只能拦截controller不能拦截到具体的方法名,这样拦截器就完全没用,经过仔细摸索,DefaultAnnotationHandlerMapping和Anno ...
- STL学习笔记(迭代器类型)
迭代器类型 迭代器是一种“能够遍历某个序列内的所有元素”的对象.它可以透过与一般指针一致的接口来完成自己的工作. 不同的迭代器具有不同的”能力“(行进和存取能力) Input迭代器 Input迭代器只 ...
- 正则化--L2正则化
请查看以下泛化曲线,该曲线显示的是训练集和验证集相对于训练迭代次数的损失. 图 1 显示的是某个模型的训练损失逐渐减少,但验证损失最终增加.换言之,该泛化曲线显示该模型与训练集中的数据过拟合.根据奥卡 ...
- PHP面试题及答案解析(6)—PHP网络编程
1.禁用COOKIE后SEESION还能用吗? 可以,COOKIE和SESSION都是用来实现会话机制的,由于http协议是无状态的,所以要想跟踪一个用户在同一个网站之间不同页面的状态,需要有这么一个 ...
- Android基础笔记(十八)- Fragment
博客的感悟终点-開始 什么是Fragment 加入fragment到Activity的两种方式 Fragment的生命周期 Fragment的向下兼容 Fragment之间的通信 博客的感悟,终点-開 ...