码迷,mamicode.com
首页 > 其他好文 > 详细

午学树链剖分有感~

时间:2019-05-02 18:56:39      阅读:190      评论:0      收藏:0      [点我收藏+]

标签:tps   node   swap   shu   图片   sign   inf   text   mod   

今天上课谈到树链剖分,被机房某dalao嘲讽了一波,决定冒着不听课的风险学链剖

关于这篇blog的灵感来源:传送门,不妥删

1.0前置知识

1.1链式向前星

这个应该都会吧,但还是附上讲解,这个图很易懂啊,蒟蒻当时一直没学懂,看了图才发现自己有多弱智

1.2dfs序

这个学长上课讲的随便找了一个讲解,仅供参考:传送门

1.3线段树

线段树应该都学了,就不附连接了

1.4为什么要学链剖

小蒟蒻没怎么做过树上倍增和链剖的题,但据说树上倍增的题链剖都能做,但链剖的题树上倍增不一定能做

2.0正文

2.1模板题传送门:洛谷p3384

2.2一些定义

  • 重儿子:父节点的所有儿子中子树结点数目最大的结点;
  • 轻儿子:父节点中除了重儿子以外的儿子;
  • 重边:父亲结点和重儿子连成的边;
  • 轻边:父亲节点和轻儿子连成的边;
  • 重链:由多条重边连接而成的路径;
  • 轻链:由多条轻边连接而成的路径;

另:

  1. 如果一个点的一堆儿子所在子树大小相等且最大,那随便找一个做重儿子,这真的很严谨
  2. 叶节点没有重儿子,非叶节点有且仅有一个重儿子

图解(手绘不易,图丑勿喷,谢谢配合)技术图片

 

黄线:重边;黑线:轻边;

重链如1-4,2-11……轻链如2-5……

红色节点:重链所在起点(top[ ]);

重儿子重边连接的节点,其余为轻儿子;

每条边的值实为dfs顺序

2.3变量声明

struct node1{//邻接表,链式向前星
    int next,to;
    #define next(i) e[i].next
    #define to(i) e[i].to
}e[maxn*2];
struct node2{//线段树维护
    int l,r,ls,rs,sum,lazy;
    #define l(i) a[i].l
    #define r(i) a[i].r
    #define sum(i) a[i].sum
    #define lazy(i) a[i].lazy
    #define ls(i) a[i].ls
    #define rs(i) a[i].rs
}a[maxn*2];
int n,m,r,rt,mod,v[maxn],head[maxn],cnt,rk[maxn];//rk存当前dfs标号在树中所对应的节点
int fa[maxn],depth[maxn],son[maxn],size[maxn];//fa他爹,depth深度,son重儿子,size子树大小(包含自身)
int top[maxn],id[maxn];//top是当前节点所在链的顶端节点(记得之前的红节点吗),id树中每个节点剖分以后的新编号(DFS的执行顺序)

酷爱宏定义的我,首先要说一下这里所有的int 都代表long long

是不是有一丝迷茫?O(∩_∩)O没关系看看下面

2.4一些函数讲解

(1)add加边操作,详见链式向前星

inline void add(int x,int y){//链式向前星加边 
    nex(++cnt)=head[x];to(cnt)=y;head[x]=cnt;
}

 

(2)dfs1,作用:处理fa,depth,size

inline void dfs1(int p){
    size[p]=1,depth[p]=depth[fa[p]]+1;//更新size,depth
    for(int i=head[x];i;i=next(i)){
        int q=to(i);
        if(q==fa[p])continue;//找到爸爸就返回
        fa[q]=p,dfs1(q),size[p]+=size[q];
        if(size[son[p]]<size[q])son[p]=q;//更新重儿子   
 }}

上图跑完以后是这个样子(这个直接截了dalao的图代表和上面一样,f代表fa,d代表depth)技术图片

(3)dfs2 作用:处理top,rk,id

inline void dfs2(int p,int tp){
    top[p]=tp,id[p]=++cnt,rk[cnt]=p;//更新数组 
    if(son[p])dfs2(son[p],tp);//有重儿子就继续递归更新下去 
    for(i=head[x];i;i=next(i)){
        int q=to(i);
        if(q!=fa[p]&&q!=son[p])
            dfs2(q,q);//显然位于轻链低端的点top为他本身 
    }
}

如果对于id和rk仍然有些不懂,请看下图

技术图片

其实rk和id就是反着的啦~

(4)pushup(p)操作用来更新以p为节点的子树的和

inline void pushup(int p){//左右子树更新和
 sum(p)=sum(ls(p))+sum(rs(p));
 sum(p)%=mod;
}

(5)build(l,r,p)建树

inline void build(int l,int r,int p){//建树,p当前节点 
    if(l==r){
        sum(p)=v[rk[l]];//或v[rk[r]] 
        l(p)=r(p)=l;return ;//或=r 
    }
    int mid=(l+r)>>1;
    ls(p)=cnt++,rs(p)=cnt++;
    build(l,mid,ls(p)),build(mid+1,r,rs(p));
    l(p)=l(ls(p)),r(p)=r(rs(p));
    pushup(p);//建了子树当然要更新一下和 
}

(6)len(p)求长度

inline int len(int p){
    return r(p)-l(p)+1;
}

(7)pushdown(p)传以p为节点的子树的懒标记

inline void pushdown(int p){//传懒标记 ,不太懂得可以先巩固一遍线段树
    if(lazy(p)){
        int ls=ls(p),rs=rs(p),lz=lazy(p);
        (lazy(ls)+=lz)%=mod;(lazy(rs)+=lz)%=mod;
        (sum(ls)+=lz*len(ls))%=mod;
        (sum(rs)+=lz*len(rs))%=mod;
        lazy(p)=0;
    }
}

(8)update(l,r,d,p)修改某一子树

inline void update(int l,int r,int d,int p){//修改某一子树 
    if(l(p)>=l&&r(p)<=r){
        lazy(p)+=d;lazy(p)%=mod;
        sum(p)+=len(p)*d;sum(p)%=mod;
        return ;
    }
    pushdown(p);
    int mid=(l(p)+r(p))>>1;
    if(mid>=l)update(l,r,d,ls(p));
    if(mid<r)update(l,r,d,rs(p));
    pushup(p);
}

(9)updates(x,y,d)从lca往下跳到x调用update再从lca跳到y调用update

inline void updates(int x,int y,int d){
    while(top[x]!=top[y]){
        if(depth[top[x]]<depth[top[y]])swap(x,y);//我们假设x更深 
        update(id[top[x]],id[x],d,root);
        x=fa[top[x]];
    }
    if(id[x]>id[y])swap(x,y);
    update(id[x],id[y],d,root);
}

(10)query(l,r,p)询问某子树和

inline int query(int l,int r,int p){ 
    if(l(p)>=l&&r(p)<=r)return sum(p);
    pushdown(p);
    int mid=(l(p)+r(p))>>1,tot=0;
    if(mid>=l)tot+=query(l,r,ls(p));
    if(mid<r)tot+=query(l,r,rs(p));
    return tot%mod;
}

(11)ask(x,y)询问路径和(ask,query和updates,update道理是一样的)

inline int ask(int x,int y){//求路径和 
    int ans=0;
    while(top[x]!=top[y]){
        if(depth[top[x]]<depth[top[y]])swap(x,y);
        ans+=query(id[top[x]],id[x],root);
        ans%=mod;
        x=fa[top[x]];
    }
    if(id[x]>id[y])swap(x,y);
    return ans+query(id[x],id[y],root)%mod;
}

(12)终于迎来了主函数!

signed main(){
    scanf("%lld%lld%lld%lld",&n,&m,&root,&mod);
    for(int i=1;i<=n;i++)scanf("%lld",&v[i]);
    for(int i=1;i<=n-1;i++){
        int x,y;
        scanf("%lld%lld",&x,&y);
        add(x,y);add(y,x);
    }
    cnt=0;dfs1(root);dfs2(root,root);
    cnt=0;build(1,n,root=cnt++);
    while(m--){
        int type;
        scanf("%lld",&type);
        switch(type){
            case 1:{
                int x,y,d;
                scanf("%lld%lld%lld",&x,&y,&d);
                updates(x,y,d);
                break;
            }
            case 2:{
                int x,y;
                scanf("%lld%lld",&x,&y);
                printf("%lld\n",ask(x,y));
                break;
            }
            case 3:{
                int x,y;
                scanf("%lld%lld",&x,&y);
                update(id[x],id[x]+size[x]-1,y,root);
                break;
            }
            case 4:{
                int x;
                scanf("%lld",&x);
                printf("%lld\n",query(id[x],id[x]+size[x]-1,root));
                break;
            }
        }
    }
    return 0;
}

3.0后记

emmmm对于这道模板题……我死了,所以在这也就不发完整代码了,大家可以借鉴一下我在开头提到的dalao的blog

如果有dalao愿意把这些函数凑到一起帮我查个错,那就感激涕零了(?????)

国际惯例:撒花完结*★,°*:.☆( ̄▽ ̄)/$:*.°★* 。

 

午学树链剖分有感~

标签:tps   node   swap   shu   图片   sign   inf   text   mod   

原文地址:https://www.cnblogs.com/klmon666/p/10802766.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!