树链剖分模板题

树链剖分的学习,我是从树链剖分-1树链剖分-2来学会的。
模板题
简单讲一下树链剖分吧,就是它首先第一遍dfs1,就将f[i] ,dep[i] ,sz[i] ,son[i]都预处理好,它们分别代表节点i的父节点,深度,子节点的数量,重节点;第二遍dfs2,就将id[i],rk[i],top[i],分别代表节点i的dfs序,dfs序为i的节点,节点i所在链的顶端节点。如果你有lca和线段树的知识,它跟lca的倍增的原理差不多,但是它每次是跳该节点的top[i]节点。然后用到线段树,根据它规定的搜素顺序(从重链开始到轻链)来建树,很简单,普通线段树怎么写就怎么写,关键是你线段树维护的是dfs序1-n每个节点对应的信息,然后它找的时候边跳边用线段树去维护。

#include<bits/stdc++.h>
#define ll long long
#define m (l+r)/2
#define ls o*2
#define rs o*2 +1
using namespace std;
const int maxn = 2e5+10;
int f[maxn],d[maxn],sz[maxn],son[maxn],top[maxn],id[maxn],rk[maxn],cnt;
vector<int>G[maxn];
ll tr[maxn*4] ,lz[maxn*4];
int n,mod,a[maxn];
void dfs1(int u,int fa,int dep)
{
    f[u] = fa;
    d[u] = dep;
    sz[u] = 1;
    for(auto v:G[u])
    {
        if(v==fa) continue;
        dfs1(v,u,dep+1);
        sz[u] += sz[v];
        if(sz[v]>sz[son[u]]||!son[u])
         son[u] = v;
    }
}
void dfs2(int u,int t)
{
    top[u] = t;
    id[u] = ++cnt;
    rk[cnt] = a[u];
    if(!son[u]) return ;
    dfs2(son[u],t);
    for(auto v:G[u])
    {
        if(v!=son[u]&&v!=f[u])
        dfs2(v,v);
    }
}
void build(int o,int l,int r)
{
    if(l==r)
    {
        tr[o] = rk[l]%mod;
        return ;
    }
    build(ls,l,m);
    build(rs,m+1,r);
    tr[o] = (tr[ls]+tr[rs])%mod;
}
void push(int o,int l,int r)
{
    if(lz[o]!=0){
     tr[ls] = (tr[ls]%mod + 1LL*(m-l+1)*lz[o]%mod)%mod ;
     tr[rs] = (tr[rs] %mod +1LL*(r-m)*lz[o]%mod)%mod;
     lz[ls] = (lz[o]+lz[ls])%mod ,  lz[rs] = (lz[rs]+lz[o])%mod;
     lz[o] = 0;
    }
}
void up(int o,int l,int r,int ql,int qr,int z)
{
    if(ql<=l&&qr>=r)
    {
      tr[o] = (tr[o]+1LL*(r-l+1)*z%mod)%mod ;
      lz[o] = (lz[o]+z%mod)%mod;
      return ;
    }
    push(o,l,r);
   if(ql<=m) up(ls,l,m,ql,qr,z);
   if(qr>m) up(rs,m+1,r,ql,qr,z);
   tr[o] = (tr[ls] + tr[rs])%mod;
}
ll qu(int o,int l,int r,int ql,int qr)
{
    if(ql<=l&&qr>=r)
    return tr[o]%mod;
    push(o,l,r);
    int ans = 0;
    if(ql<=m) ans =(ans+qu(ls,l,m,ql,qr))%mod;
    if(qr>m) ans = (ans+qu(rs,m+1,r,ql,qr))%mod;
    return ans%mod ;
}
void up_path(int x,int y,int z)
{
    int fx = top[x] , fy =top[y];
    while(fx!=fy)
    {
        if(d[fx]>d[fy]){
            up(1,1,cnt,id[fx],id[x],z);
            x = f[fx];
        }
        else {
           up(1,1,cnt,id[fy],id[y],z);
           y = f[fy];
        }
        fx = top[x] , fy = top[y];
    }
    if(id[x]<=id[y]) up(1,1,cnt,id[x],id[y],z);
    else up(1,1,cnt,id[y],id[x],z);
}
ll qu_path(int x,int y)
{
    ll ans = 0;
    int fx = top[x] , fy = top[y];
    while(fx!=fy)
    {
        if(d[fx]>d[fy])
        {
            ans = (ans + qu(1,1,cnt,id[fx],id[x]))%mod;
            x = f[fx];
        }
        else {
            ans = (ans+qu(1,1,cnt,id[fy],id[y]))%mod;
            y = f[fy];
        }
        fx = top[x] , fy = top[y];
    }
     if(id[x]<=id[y])  ans = (ans+qu(1,1,cnt,id[x],id[y]))%mod;
     else ans = (ans+qu(1,1,cnt,id[y],id[x]))%mod;
    return ans%mod ;
}
int main()
{
    int u,v;
    int q,rt;scanf("%d%d%d%d",&n,&q,&rt,&mod);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<n;i++)
    {
     scanf("%d%d",&u,&v);
     G[u].push_back(v);
     G[v].push_back(u);
    }
    dfs1(rt,0,1);
    dfs2(rt,rt);
    build(1,1,n);
    int op,x,y,z;
    for(int i=1;i<=q;i++)
    {
      scanf("%d",&op);
      if(op==1){
        scanf("%d%d%d",&x,&y,&z);
        up_path(x,y,z%mod);
      }
      if(op==2){
        scanf("%d%d",&x,&y);
         printf("%lld\n",qu_path(x,y));
      }
      if(op==3)
      {
          scanf("%d%d",&x,&z);
          up(1,1,cnt,id[x],id[x]+sz[x]-1,z%mod);
      }
      if(op==4)
      {
          scanf("%d",&x);
          printf("%lld\n",qu(1,1,cnt,id[x],id[x]+sz[x]-1));
      }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值