树链剖分

适用题目特征

修改树上两点之间的路径上/子树中所有点的值。
查询树上两点之间的路径上/子树中节点权值的和/极值/其它(在序列上可以用数据结构维护,便于合并的信息)

原理

一颗子树/链内的dfs序是连续的。所以可以用数据结构维护。

例题

Luogu P3384
代码如下

/*
Luogu P3384
*/
#define method_1
#ifdef method_1
/*

*/
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iomanip>
#include<ctime>
#include<string>
#include<bitset>
#define D(x) cout<<#x<<" = "<<x<<"  "
#define E cout<<endl
#define rep(i,s,t) for(int i=(s);i<=(t);i++)
#define rep0(i,s,t) for(int i=(s);i<(t);i++)
#define rrep(i,t,s) for(int i=(t);i>=(s);i--)
using namespace std;
typedef long long ll;
typedef pair<int,int>pii;
const int maxn=1e5+5;
const int INF=0x3f3f3f3f;
int n,m,root,mod;
struct node{
    int from,to;
}edge[maxn<<1];
int head[maxn],tot=1;
int w[maxn];
void add(int from,int to){
    edge[++tot].from=head[from],head[from]=tot,edge[tot].to=to;
}
int f[maxn],d[maxn],sz[maxn],son[maxn];
int top[maxn],wt[maxn],id[maxn],cnt=0;
void dfs1(int x,int fa,int deep){
    d[x]=deep,sz[x]=1,f[x]=fa;
    int maxs=0;
    for(int i=head[x];i;i=edge[i].from){
        int y=edge[i].to;
        if(y==fa) continue;
        dfs1(y,x,deep+1);
        sz[x]+=sz[y];
        if(sz[y]>maxs) {maxs=sz[y],son[x]=y;}
    }
}
void dfs2(int x,int anc){
    id[x]=++cnt,wt[cnt]=w[x],top[x]=anc;
    if(!son[x]) return;
    dfs2(son[x],anc);
    for(int i=head[x];i;i=edge[i].from){
        int y=edge[i].to;
        if(y==f[x]||y==son[x]) continue;
        dfs2(y,y);
    }
}
struct segmentTree{
    int l,r,sum,tag,sz;
}tr[maxn<<2];
void build(int rt,int l,int r){
    tr[rt].l=l,tr[rt].r=r,tr[rt].sz=r-l+1;
    if(l==r){tr[rt].sum=wt[l];return;}
    int mid=l+r>>1;
    build(rt<<1,l,mid);build(rt<<1|1,mid+1,r);
    tr[rt].sum=(tr[rt<<1].sum+tr[rt<<1|1].sum+mod)%mod;
}
void spread(int rt){
    if(tr[rt].tag){
        tr[rt<<1].tag=(tr[rt<<1].tag+tr[rt].tag)%mod;
        tr[rt<<1|1].tag=(tr[rt<<1|1].tag+tr[rt].tag)%mod;
        tr[rt<<1].sum=(tr[rt<<1].sum+tr[rt<<1].sz*tr[rt].tag)%mod;
        tr[rt<<1|1].sum=(tr[rt<<1|1].sum+tr[rt<<1|1].sz*tr[rt].tag)%mod;
        tr[rt].tag=0;
    }
}
void update(int rt,int l,int r,int v){
    if(tr[rt].l>=l&&tr[rt].r<=r){tr[rt].sum=(tr[rt].sum+tr[rt].sz*v)%mod;tr[rt].tag=(tr[rt].tag+v)%mod;return;}
    spread(rt);
    int mid=tr[rt].l+tr[rt].r>>1;
    if(mid>=l) update(rt<<1,l,r,v);
    if(mid<r) update(rt<<1|1,l,r,v);
    tr[rt].sum=(tr[rt<<1].sum+tr[rt<<1|1].sum+mod)%mod;
}
int query(int rt,int l,int r){
    if(tr[rt].l>=l&&tr[rt].r<=r) return tr[rt].sum%mod;
    spread(rt);
    int mid=tr[rt].l+tr[rt].r>>1;
    int res=0;
    if(mid>=l) res=(res+query(rt<<1,l,r))%mod;
    if(mid<r) res=(res+query(rt<<1|1,l,r))%mod;
    return res;
}
void treeAdd(int x,int y,int v){
    while(top[x]!=top[y]){
        if(d[top[x]]<d[top[y]]) swap(x,y);
        update(1,id[top[x]],id[x],v);
        x=f[top[x]];
    }
    if(d[x]>d[y]) swap(x,y);
    update(1,id[x],id[y],v);
}
int treeSum(int x,int y){
    int res=0;
    while(top[x]!=top[y]){
        if(d[top[x]]<d[top[y]]) swap(x,y);
        res=(res+query(1,id[top[x]],id[x]))%mod;
        x=f[top[x]];
    }
    if(d[x]>d[y]) swap(x,y);
    res=(res+query(1,id[x],id[y]))%mod;
    return res;
}
void solve(){
    dfs1(root,0,1);
    dfs2(root,root);
    build(1,1,n);
//  rep(i,1,n) D(wt[i]);
//  E; 
    while(m--){
        int op,x,y,z;
        scanf("%d",&op);
        if(op==1){
            scanf("%d%d%d",&x,&y,&z);
            treeAdd(x,y,z%mod); 
        }
        if(op==2){
            scanf("%d%d",&x,&y);
            printf("%d\n",treeSum(x,y));
        }
        if(op==3){
            scanf("%d%d",&x,&z);
            update(1,id[x],id[x]+sz[x]-1,z%mod);
        }
        if(op==4){
            scanf("%d",&x);
            printf("%d\n",query(1,id[x],id[x]+sz[x]-1));
        }
    }
}
int main() {
//  ios::sync_with_stdio(false);
//  freopen("轻重链剖分.in","r",stdin);
    scanf("%d%d%d%d",&n,&m,&root,&mod);
    rep(i,1,n) scanf("%d",&w[i]);
    int from,to;
    rep(i,1,n-1) scanf("%d%d",&from,&to),add(from,to),add(to,from);
    solve();
    return 0;
}
#endif
#ifdef method_2

#endif
#ifdef method_3
/*

*/

#endif
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 227,156评论 6 529
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 97,866评论 3 413
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 174,880评论 0 373
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 62,398评论 1 308
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 71,202评论 6 405
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 54,743评论 1 320
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 42,822评论 3 438
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 41,962评论 0 285
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 48,476评论 1 331
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 40,444评论 3 354
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 42,579评论 1 365
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 38,129评论 5 355
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 43,840评论 3 344
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 34,231评论 0 25
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 35,487评论 1 281
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 51,177评论 3 388
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 47,568评论 2 370

推荐阅读更多精彩内容

  • LUOGU 3384Description如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需...
    苏子旃阅读 309评论 0 0
  • 一般来讲,我们很熟悉序列的操作,对于树上路径的操作会觉得比较棘手。而树链剖分基本思想,是将树上的路径问题转化为序列...
    Catzyzy阅读 850评论 0 2
  • 题目链接 https://www.luogu.org/problem/P3384 分析 树链剖分是将树适当划分成链...
    Mr94Kevin阅读 308评论 0 0
  • 学习树链剖分我看过以下博客:树链剖分原理和实现树链剖分整理总结 知道大概之后,我以为要多加深记忆的地方:对于每一个...
    无令便逐风阅读 711评论 0 1
  • 黑色的海岛上悬着一轮又大又圆的明月,毫不嫌弃地把温柔的月色照在这寸草不生的小岛上。一个少年白衣白发,悠闲自如地倚坐...
    小水Vivian阅读 3,123评论 1 5