洛谷树剖模板题 P3384 | 树链剖分

原题链接

对于以u为根的子树,后代节点的dfn显然比他的dfn大,我们可以记录一下回溯到u的dfn,显然这两个dfn构成了一个连续区间,代表u及u的子树

剩下的就和树剖一样了

#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 100010
typedef long long ll;
using namespace std;
int ecnt,head[N],son[N],fa[N],sz[N],n,m,r,deep[N],pos[N],indx[N],tot,op,back[N],top[N];
ll val[N],P;
struct adj
{
    int nxt,v;
}e[2*N];
struct node
{
    int l,r;
    ll sum,lz;
}t[4*N];
void add(int u,int v)
{
    e[++ecnt].v=v;
    e[ecnt].nxt=head[u];
    head[u]=ecnt;
    e[++ecnt].v=u;
    e[ecnt].nxt=head[v];
    head[v]=ecnt;
}
void dfs1(int x,int father,int dep)
{
    deep[x]=dep,fa[x]=father,sz[x]=1;
    for (int i=head[x];i;i=e[i].nxt)
    {
    int v=e[i].v;
    if (fa[x]==v) continue;
    dfs1(v,x,dep+1);
    sz[x]+=sz[v];
    if (sz[v]>sz[son[x]]) son[x]=v;
    }
}
void dfs2(int x,int TOP)
{
    top[x]=TOP;
    pos[x]=++tot;
    indx[tot]=x;
    if (son[x]) dfs2(son[x],TOP);
    for (int i=head[x];i;i=e[i].nxt)
    {
    int v=e[i].v;
    if (v==fa[x] || v==son[x]) continue;
    dfs2(v,v);
    }
    back[x]=tot;
}
void pushdown(int p)
{
    if (t[p].l==t[p].r || !t[p].lz) return;
    int w=t[p].lz;
    t[p<<1].sum=(t[p<<1].sum+w*(t[p<<1].r-t[p].l+1)%P)%P;
    t[p<<1|1].sum=(t[p<<1|1].sum+w*(t[p<<1|1].r-t[p<<1|1].l+1)%P)%P;
    t[p<<1].lz+=w;
    t[p<<1].lz%=P;
    t[p<<1|1].lz+=w;
    t[p<<1|1].lz%=P;
    t[p].lz=0;
}
void pushup(int p)
{
    t[p].sum=(t[p<<1].sum+t[p<<1|1].sum)%P;
}
void build(int p,int l,int r)
{
    t[p].l=l,t[p].r=r,t[p].lz=0;
    if (l!=r)
    {
    int mid=l+r>>1;
    build(p<<1,l,mid);
    build(p<<1|1,mid+1,r);
    pushup(p);
    }
    else
    t[p].sum=val[indx[l]]%P;
}
void modify(int p,int l,int r,int w)
{
    if (t[p].l==l && t[p].r==r)
    {
    t[p].sum+=w*(t[p].r-t[p].l+1);
    t[p].lz+=w;
    return ;
    }
    pushdown(p);
    int mid=t[p].l+t[p].r>>1;
    if (r<=mid) modify(p<<1,l,r,w);
    else if (l>mid) modify(p<<1|1,l,r,w);
    else modify(p<<1,l,mid,w),modify(p<<1|1,mid+1,r,w);
    pushup(p);
}
ll query(int p,int l,int r)
{
    if (t[p].l==l && t[p].r==r)
    return t[p].sum%P;
    int mid=t[p].l+t[p].r>>1;
    pushdown(p);
    if (r<=mid) return query(p<<1,l,r)%P;
    if (l>mid) return query(p<<1|1,l,r)%P;
    return (query(p<<1,l,mid)+query(p<<1|1,mid+1,r))%P;
}
void pathInc(int u,int v,int w)
{
    while (top[u]!=top[v])
    {
    if (deep[top[u]]<deep[top[v]]) swap(u,v);
    modify(1,pos[top[u]],pos[u],w);
    u=fa[top[u]];
    }
    if (deep[u]>deep[v]) swap(u,v);
    modify(1,pos[u],pos[v],w);
}
ll pathQuery(int u,int v)
{
    ll ret=0;
    while (top[u]!=top[v])
    {
    if (deep[top[u]]<deep[top[v]]) swap(u,v);
    ret=(ret+query(1,pos[top[u]],pos[u]))%P;
    u=fa[top[u]];
    }
    if (deep[u]>deep[v]) swap(u,v);
    return (ret+query(1,pos[u],pos[v]))%P;
}
int main()
{
    scanf("%d%d%d%lld",&n,&m,&r,&P);
    for (int i=1;i<=n;i++)
    scanf("%lld",&val[i]);
    for (int i=1,u,v;i<n;i++)
    scanf("%d%d",&u,&v),add(u,v);
    dfs1(r,0,0);
    dfs2(r,r);
    build(1,1,n);
    for (int i=1,x,y,z;i<=m;i++)
    {
    scanf("%d",&op);
    if (op==1)
        scanf("%d%d%d",&x,&y,&z),pathInc(x,y,z);
    else if (op==2)
        scanf("%d%d",&x,&y),printf("%lld
",pathQuery(x,y));
    else if (op==3)
        scanf("%d%d",&x,&z),modify(1,pos[x],back[x],z);
    else scanf("%d",&x),printf("%lld
",query(1,pos[x],back[x]));
    }
    return 0;
}