模板_树链剖分

#define lc (o<<1)
#define rc ((o<<1)|1)
#define m ((l+r)>>1)
struct Edge{int to,next;}edge[N*2];
int head[N],cnt;//邻接表 
int top[N],deep[N],son[N],f[N],size[N],s[N];//top表示重链顶点,deep表示深度,son表示重儿子,f表示父节点,size表示子节点数量,s表示权值 
int w[N],p[N],pos;//w表示与父节点连边在线段树中的位置,p表示线段树中节点的原节点编号,线段树节点数 
int n;
inline void init(){
    memset(head,-1,sizeof(head));cnt=0,pos=0;
}
inline void AddEdge(int fr,int to){edge[cnt].to=to,edge[cnt].next=head[fr],head[fr]=cnt++;}
inline void dfs1(int u,int pre,int d){//第一遍求f,deep,son,size 
    f[u]=pre,deep[u]=d,size[u]=1;
    for(int i=head[u],v=edge[i].to;i!=-1;i=edge[i].next,v=edge[i].to)
        if(v!=pre){
            dfs1(v,u,d+1);size[u]+=size[v];
            if(!son[u]||size[v]>size[son[u]]) son[u]=v;
        }
}
inline void dfs2(int u,int t){//第二遍求重儿子并编号 
    top[u]=t,w[u]=++pos,p[w[u]]=u;
    if(!son[u]) return;dfs2(son[u],t);
    for(int i=head[u],v=edge[i].to;i!=-1;i=edge[i].next,v=edge[i].to)
        if(v!=son[u]&&v!=f[u]) dfs2(v,v);
}
struct Node{int sum,ma;}node[N*4];//节点
inline void updata(int o){node[o].ma=max(node[lc].ma,node[rc].ma);
    node[o].sum=node[lc].sum+node[rc].sum;}
inline void Build(int o,int l,int r){
    if(l==r){node[o].ma=node[o].sum=s[p[l]];return;}
    Build(lc,l,m);Build(rc,m+1,r);updata(o);
}
inline void Modify(int o,int l,int r,int k,int v){
    if(l==r){node[o].sum=node[o].ma=v;return;}
    if(k<=m) Modify(lc,l,m,k,v);else Modify(rc,m+1,r,k,v);
    updata(o);return;
}
int Max(int o,int l,int r,int L,int R){
    if(L<=l&&r<=R) return node[o].ma;int ans=INT_MIN;
    if(L<=m) ans=max(ans,Max(lc,l,m,L,R));if(R>m) ans=max(ans,Max(rc,m+1,r,L,R));
    return ans;
}
int QueryMax(int u,int v){
    int f1=top[u],f2=top[v],ans=INT_MIN;
    while(f1!=f2){
        if(deep[f1]<deep[f2]) swap(u,v),swap(f1,f2);
        ans=max(ans,Max(1,1,n,w[f1],w[u]));u=f[f1],f1=top[u];
    }
    ans=max(ans,(deep[u]>deep[v])?Max(1,1,n,w[v],w[u]):Max(1,1,n,w[u],w[v]));
    return ans;
}
int Sum(int o,int l,int r,int L,int R){
    if(L<=l&&r<=R) return node[o].sum;int ans=0;
    if(L<=m) ans+=Sum(lc,l,m,L,R);if(R>m) ans+=Sum(rc,m+1,r,L,R);
    return ans;
}
int QuerySum(int u,int v){
    int f1=top[u],f2=top[v],ans=0;
    while(f1!=f2){
        if(deep[f1]<deep[f2]) swap(u,v),swap(f1,f2);
        ans+=Sum(1,1,n,w[f1],w[u]);u=f[f1],f1=top[u];
    }
    ans+=(deep[u]>deep[v])?Sum(1,1,n,w[v],w[u]):Sum(1,1,n,w[u],w[v]);
    return ans;
}