标签：ch 剖分 int top son ans 树链 id
树链剖分
前置芝士
就像它的名字，树链剖分是在一棵树上进行，在讲解中还会用到线段树和dfs，如果不会，打开链接自行搜索（主要是线段树的博客没做，还有不要问我为什么这算知识）。
一个节点的重儿子，为其更大的一颗子树的根节点。从这个点连向重儿子的边我们称为重边。
由重边连续连起来的点和边就组成了重链，也就是树链。
概念
将一棵树划分成若干条链，用数据结构去维护每条链，复杂度为 \(O(logN)\) 。
其实本质是一些数据结构/算法在树上的推广
作用
处理树上的一些相关问题。比如——维护树上区间，树上路径等等。
区间我们想到了线段树，树上路径想到了LCA，但是它们都有一个特点——连续。线段树只能维护连续区间，LCA路径也是不间断的。所以为了便于处理，我们要对这个图重新标号，以便查找。怎么标呢？我们可以想到——在树链上操作LCA路径，那么路径也是要连贯的，也就是说重链上的编号要连贯，所以我们重新编号的时候是在dfs序的基础上遵循先遍历重儿子的原则。
例题
思路
可以很容易发现——操作1、2都需要走一遍x到y的路径，操作3、4都需要操作以x为根的子树。所以我们先思考怎么遍历这些区间——
首先遍历x到y的路径，我们亦容易想到LCA——两个点同时往上跳，直到某个值相同，可以一起操作。所以我们的思路就是：两个点不在同一条链就往链头的父亲节点跳，在同一条链上就直接处理。而处理方法也很简单——因为全程都在链上以连续的新节点编号来操作，所以线段树维护区间距离就很方便了，完全不受树剖影响地敲一个基本的建树、查询、区间修改+延迟标记的代码就可以了。
而对于操作3和4，以x为根的子树，显然编号也是连续的——毕竟编号时的最基本原则还是dfs遍历。但是有一个小问题——我们知道以x为根的子树最小的编号是x的编号，但是最大的编号我们并不知道，如果遍历一遍来找的话复杂度就会比较高了。所以——这是我们在初始化树剖的时候要存储下来的一个变量——以x为根的子树的最大的节点编号。也就是代码中的son[ ]。
代码
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#define Rint register int
#define mem(a,b) memset(a,(b),sizeof(a))
#define Temp template<typename T>
using namespace std;
typedef long long LL;
Temp inline void read(T &x){
x=0;T w=1,ch=getchar();
while(!isdigit(ch)&&ch!='-')ch=getchar();
if(ch=='-')w=-1,ch=getchar();
while(isdigit(ch))x=(x<<3)+(x<<1)+(ch^'0'),ch=getchar();
x=x*w;
}
#define mid ((l+r)>>1)
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define len (r-l+1)
const int maxn=200000+10;
int n,m,r,mod;
//见题意
int e,beg[maxn],nex[maxn],to[maxn],w[maxn],wt[maxn];
//链式前向星数组，w[]、wt[]初始点权数组
int a[maxn<<2],laz[maxn<<2];
//线段树数组、lazy操作
int son[maxn],id[maxn],fa[maxn],cnt,dep[maxn],siz[maxn],top[maxn];
//son[]重儿子编号,id[]新编号,fa[]父亲节点,cnt dfs_cl ock/dfs序,dep[]深度,siz[]子树大小,top[]当前链顶端节点
int res=0;
//查询答案
inline void add(int x,int y){//链式前向星加边
to[++e]=y;
nex[e]=beg[x];
beg[x]=e;
}
//-------------------------------------- 以下为线段树
inline void pushdown(int rt,int lenn){
laz[rt<<1]+=laz[rt];
laz[rt<<1|1]+=laz[rt];
a[rt<<1]+=laz[rt]*(lenn-(lenn>>1));
a[rt<<1|1]+=laz[rt]*(lenn>>1);
a[rt<<1]%=mod;
a[rt<<1|1]%=mod;
laz[rt]=0;
}
inline void build(int rt,int l,int r){
if(l==r){
a[rt]=wt[l];
if(a[rt]>mod)a[rt]%=mod;
return;
}
build(lson);
build(rson);
a[rt]=(a[rt<<1]+a[rt<<1|1])%mod;
}
inline void query(int rt,int l,int r,int L,int R){
if(L<=l&&r<=R){res+=a[rt];res%=mod;return;}
else{
if(laz[rt])pushdown(rt,len);
if(L<=mid)query(lson,L,R);
if(R>mid)query(rson,L,R);
}
}
inline void update(int rt,int l,int r,int L,int R,int k){
if(L<=l&&r<=R){
laz[rt]+=k;
a[rt]+=k*len;
}
else{
if(laz[rt])pushdown(rt,len);
if(L<=mid)update(lson,L,R,k);
if(R>mid)update(rson,L,R,k);
a[rt]=(a[rt<<1]+a[rt<<1|1])%mod;
}
}
//---------------------------------以上为线段树
inline int qRange(int x,int y){
int ans=0;
while(top[x]!=top[y]){//当两个点不在同一条链上
if(dep[top[x]]<dep[top[y]])swap(x,y);//把x点改为所在链顶端的深度更深的那个点
res=0;
query(1,1,n,id[top[x]],id[x]);//ans加上x点到x所在链顶端 这一段区间的点权和
ans+=res;
ans%=mod;//按题意取模
x=fa[top[x]];//把x跳到x所在链顶端的那个点的上面一个点
}
//直到两个点处于一条链上
if(dep[x]>dep[y])swap(x,y);//把x点改为所在链顶端的深度更浅的那个点
res=0;
query(1,1,n,id[x],id[y]);//这时再加上此时两个点的区间和即可
ans+=res;
return ans%mod;
}
inline void updRange(int x,int y,int k){//同上
k%=mod;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
update(1,1,n,id[top[x]],id[x],k);
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
update(1,1,n,id[x],id[y],k);
}
inline int qSon(int x){
res=0;
query(1,1,n,id[x],id[x]+siz[x]-1);//子树区间右端点为id[x]+siz[x]-1
return res;
}
inline void updSon(int x,int k){//同上
update(1,1,n,id[x],id[x]+siz[x]-1,k);
}
inline void dfs1(int x,int f,int deep){//x当前节点，f父亲，deep深度
dep[x]=deep;//标记每个点的深度
fa[x]=f;//标记每个点的父亲
siz[x]=1;//标记每个非叶子节点的子树大小
int maxson=-1;//记录重儿子的儿子数
for(Rint i=beg[x];i;i=nex[i]){
int y=to[i];
if(y==f)continue;//若为父亲则continue
dfs1(y,x,deep+1);//dfs其儿子
siz[x]+=siz[y];//把它的儿子数加到它身上
if(siz[y]>maxson)son[x]=y,maxson=siz[y];//标记每个非叶子节点的重儿子编号
}
}
inline void dfs2(int x,int topf){//x当前节点，topf当前链的最顶端的节点
id[x]=++cnt;//标记每个点的新编号
wt[cnt]=w[x];//把每个点的初始值赋到新编号上来
top[x]=topf;//这个点所在链的顶端
if(!son[x])return;//如果没有儿子则返回
dfs2(son[x],topf);//按先处理重儿子，再处理轻儿子的顺序递归处理
for(Rint i=beg[x];i;i=nex[i]){
int y=to[i];
if(y==fa[x]||y==son[x])continue;
dfs2(y,y);//对于每一个轻儿子都有一条从它自己开始的链
}
}
int main(){
read(n);read(m);read(r);read(mod);
for(Rint i=1;i<=n;i++)read(w[i]);
for(Rint i=1;i<n;i++){
int a,b;
read(a);read(b);
add(a,b);add(b,a);
}
dfs1(r,0,1);
dfs2(r,r);
build(1,1,n);
while(m--){
int k,x,y,z;
read(k);
if(k==1){
read(x);read(y);read(z);
updRange(x,y,z);
}
else if(k==2){
read(x);read(y);
printf("%d\n",qRange(x,y));
}
else if(k==3){
read(x);read(y);
updSon(x,y);
}
else{
read(x);
printf("%d\n",qSon(x));
}
}
}
例题2
思路
这是边转点，顾名思义就是把边权转化为点权来用。而又因为每个点的父亲是唯一的，所以我们都是把边权给了深度更大的那个点。
在这个题里，我们用边转点的树链剖分来求LCA，剩下的和本题题解思路一样，用Kruskal求最大生成树。。。
代码
#include<bits/stdc++.h>
#define maxn 10005
#define maxm 50005
using namespace std;
int read()
{
register int x = 0, f = 1, ch = getchar();
while(!isdigit(ch)) {if(ch == '-') f = -1; ch = getchar();}
while(isdigit(ch)) x = (x << 1) + (x << 3) + ch - '0', ch = getchar();
return x * f;
}
int n, m;
struct node
{
int u, v, w;
bool operator < (const node &x) const
{
return x.w < w;
}
}g[maxm];
int fa[maxn];
int get(int x)
{
if(fa[x] == x) return x;
return fa[x] = get(fa[x]);
}
struct edge
{
int to, w, nxt;
edge(){}
edge(int tt, int ww, int nn)
{
to = tt, w = ww, nxt = nn;
}
}e[maxn << 1];
int k = 0, head[maxn];
void add(int u, int v, int w)
{
e[k] = edge(v, w, head[u]);
head[u] = k++;
}
int dep[maxn], size[maxn], son[maxn], val[maxn], fa_[maxn];
void dfs_1(int u)//树剖初始化1
{
size[u] = 1;
for(register int i = head[u]; ~i; i = e[i].nxt)
{
register int v = e[i].to, w = e[i].w;
if(v == fa_[u]) continue;
dep[v] = dep[u] + 1;
fa_[v] = u;
dfs_1(v);
size[u] += size[v];
if(size[v] > size[son[u]]) son[u] = v;
}
}
int top[maxn], dfn[maxn], tot = 0;
void dfs_2(int u, int tp)//树剖初始化2
{
top[u] = tp;
dfn[u] = ++tot;
if(son[u]) dfs_2(son[u], tp);
for(register int i = head[u]; ~i; i = e[i].nxt)
{
register int v = e[i].to, w = e[i].w;
if(v != fa_[u] && v != son[u]) dfs_2(v, v);
val[dfn[v]] = w;
}
}
int road[maxn << 2];
void build(int p, int l, int r)//线段树建树
{
if(l == r)
{
road[p] = val[l];
return;
}
register int mid = l + r >> 1;
build(p << 1, l, mid);
build(p << 1 | 1, mid + 1, r);
road[p] = min(road[p << 1], road[p << 1 | 1]);
}
int ask(int p, int l, int r, int ls, int rs)//线段树查询
{
if(ls <= l && r <= rs)
return road[p];
}
register int mid = l + r >> 1, ans = 1 << 30;
if(ls <= mid) ans = min(ans, ask(p << 1, l, mid, ls, rs));
if(rs > mid) ans = min(ans, ask(p << 1 | 1, mid + 1, r, ls, rs));
return ans;
}
void work(int u, int v)//树剖基本操作
{
register int ans = 1 << 30;
register int tmp1 = u, tmp2 = v;
while(top[u] != top[v])
{
if(dep[top[u]] > dep[top[v]]) swap(u, v);
ans = min(ans, ask(1, 1, n, dfn[top[v]], dfn[v]));
v = fa_[top[v]];
}
if(dep[u] > dep[v]) swap(u, v);
ans = min(ans, ask(1, 1, n, dfn[son[u]], dfn[v]));
printf("%d\n", ans);
}
int main()
{
n = read(), m = read();
for(register int i = 1; i <= n; i++)
fa[i] = i;
for(register int i = 1; i <= m; i++)
g[i].u = read(), g[i].v = read(), g[i].w = read();
sort(g + 1, g + 1 + m);
memset(head, -1, sizeof head);
register int cnt = 0;//最大生成树开始
for(register int i = 1; i <= m; i++)
{
register int u = g[i].u, v = g[i].v, w = g[i].w;
if(get(u) == get(v)) continue;
fa[get(u)] = get(v);
add(u, v, w);
add(v, u, w);
cnt++;
if(cnt == n - 1) break;//省时处理
}
for(int i = 1; i <= n; i++)//这里一定要开for！！！QAQ
if(!size[i]) dfs_1(i);
for(int i = 1; i <= n; i++)//这里也一定要开for！！！QAQ
if(!top[i]) dfs_2(i, i);
val[1] = 1 << 30;//赋值
build(1, 1, n);
register int q, u, v;
q = read();
while(q--)
{
u = read(), v = read();
if(get(u) != get(v))//判定不连通情况
{
puts("-1");continue;
}
work(u, v);
}
}
标签：ch,剖分,int,top,son,ans,树链,id 来源： https://www.cnblogs.com/jasony/p/13339526.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。