【图论】树链剖分

树链剖分详解 - 自为风月马前卒 - 博客园 (cnblogs.com)

P3384 【模板】重链剖分/树链剖分 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include<iostream>
using namespace std;


void dfs1(int u,int father){
fa[u]=father;
dep[u]=dep[father]+1;
sz[u]=1;




for(int i=head[u];~i;i=edge[i].next){int v=edge[i].to;s


    if(v==father)continue;
    dfs1(v,u);
    if(sz[son[u]<=sz[v])son[u]=v;
}


}


//每个dfs完成3个初始化
void dfs2(int u,int t){//重编号
    top[u]=t;
    idx[u]=++cnt;
    a[cnt]=b[u];


    if(!son[u])return ;
    dfs(son[u],t);//重儿子链首相同

    for(int i=head[u];~i;i=edge[i].next){        int v=edge[i].to;
            if(!idx[v])//现在idx是独一无二的时间戳
            {
                dfs(v,v);
            }
    }

}





struct tree{
int l,r,w,add,sum;
}tr[];
void bui(int id,int l,int r){
tr[id]={l,r,0,a[r]};
if(l==r)return ;

bui(ls,l,mid);
bui(rs,mid+1,r);
pushup(id);

}

void pushup(int id){
tr[id].sum+=tr[ls].sum+tr[rs].sum;
}

void pushdown(int id){
//add看的不好的话用lazy
if(tr[id].add){
    tr[ls].sum+=tr[id].add*(tr[ls].r-tr[ls].l+1);
    tr[rs].sum+=tr[id].add*(tr[rs].r-tr[rs].l+1);

    tr[ls].add+=tr[id].add;
    tr[rs].add+=tr[id].add;

    tr[id]=0;
}

}




ll query(int id,int x,int y){
if(x<=tr[id].l&&y>=tr[id].r){return tr[id].sum;}
pushdown(id);


ll res=0;
if(l<=mid)res+=query(ls,x,y);
if(r>=mid+1)res+=query(rs,x,y);

return res;

}


ll querypath(int u,int v){

while(top[u]!=top[v]){
    if(dep[top[u]]<=top[v])swap(u,v);

    res+=query(1,idx[top[u]],idx[u]);
    u=fa[top[u]];
}


if(dep[u]<dep[v])swap(u,v);
res+=query(1,idx[v],idx[u]);

return res;

}



void update(int id,int x,int y,int k){
if(x<=tr[id].l&&y>=tr[id].r){tr[id].add+=k;tr[id].sum+=k*(tr[id].r-tr[id].l+1);return ;}


pushdown(id);
if(x<=mid)update(ls,x,y,k);
if(y>=mid+1)update(rs,x,y,k);
pushup(id);

}

void updatepath(int u,int v,int k){

while(top[u]!=top[v]){
    if(dep[top[u]]<dep[top[v]])swap(u,v);

    update(idx[top[u]],idx[u],k);
    u=fa[top[u]];
}

if(dep[u]<dep[v])swap(u,v);
update(idx[u],idx[v],k);
}