标签：剖分 int siz tr 树链 maxn res id 模板

题目链接：https://www.luogu.org/problemnew/show/P3384

题意：树链剖分模板题，但是比较坑的是要注意取模，每个可能炸int的地方都要加取模。

代码如下：

#include<cstdio>
#include<algorithm>
using namespace std;

const int maxn=100005;
//链式向前星
int cnt1,head[maxn],w[maxn],wt[maxn];
int n,m,r,Mod,res;
int son[maxn],id[maxn],fa[maxn],cnt2,dep[maxn],siz[maxn],top[maxn];

struct node1{
    int v,nex;
}edge[maxn<<1];

void adde(int u,int v){
    edge[++cnt1].v=v;
    edge[cnt1].nex=head[u];
    head[u]=cnt1;
}
//线段树部分
struct node2{
    int l,r,val,add;
}tr[maxn<<2];

void build(int v,int l,int r){
    tr[v].l=l,tr[v].r=r;
    if(l==r){
        tr[v].val=wt[r]%Mod;
        return;
    }
    int mid=(l+r)>>1;
    build(v<<1,l,mid);
    build(v<<1|1,mid+1,r);
    tr[v].val=(tr[v<<1].val+tr[v<<1|1].val+Mod)%Mod;
}

void pushdown(int v){
    tr[v<<1].val+=tr[v].add*(tr[v<<1].r-tr[v<<1].l+1);
    tr[v<<1|1].val+=tr[v].add*(tr[v<<1|1].r-tr[v<<1|1].l+1);
    tr[v<<1].val%=Mod;
    tr[v<<1|1].val%=Mod;
    tr[v<<1].add+=tr[v].add;
    tr[v<<1|1].add+=tr[v].add;
    tr[v].add=0;
}

void update(int v,int l,int r,int k){
    if(l<=tr[v].l&&r>=tr[v].r){
        tr[v].val+=k*(tr[v].r-tr[v].l+1);
        tr[v].val%=Mod;
        tr[v].add+=k;
        return;
    }
    if(tr[v].add) pushdown(v);
    int mid=(tr[v].l+tr[v].r)>>1;
    if(l<=mid) update(v<<1,l,r,k);
    if(r>mid) update(v<<1|1,l,r,k);
    tr[v].val=(tr[v<<1].val+tr[v<<1|1].val+Mod)%Mod;
}

void query(int v,int l,int r){
    if(l<=tr[v].l&&r>=tr[v].r){
        res+=tr[v].val;
        res%=Mod;
        return;
    }
    if(tr[v].add) pushdown(v);
    int mid=(tr[v].l+tr[v].r)>>1;
    if(l<=mid) query(v<<1,l,r);
    if(r>mid) query(v<<1|1,l,r);
    tr[v].val=(tr[v<<1].val+tr[v<<1|1].val+Mod)%Mod;
}

int qRange(int x,int y){
    int ans=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        res=0;
        query(1,id[top[x]],id[x]);
        ans=(ans+res)%Mod;
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    res=0;
    query(1,id[x],id[y]);
    ans=(ans+res)%Mod;
    return ans;
}

void updRange(int x,int y,int k){
    k%=Mod;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        update(1,id[top[x]],id[x],k);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    update(1,id[x],id[y],k);
}

int qSon(int x){
    res=0;
    query(1,id[x],id[x]+siz[x]-1);
    return res;
}

void updSon(int x,int k){
    update(1,id[x],id[x]+siz[x]-1,k);
}

void dfs1(int x,int f,int deep){
    dep[x]=deep;
    fa[x]=f;
    siz[x]=1;
    int maxson=-1;
    for(int i=head[x];i;i=edge[i].nex){
        int y=edge[i].v;
        if(y==f) continue;
        dfs1(y,x,deep+1);
        siz[x]+=siz[y];
        if(siz[y]>maxson) son[x]=y,maxson=siz[y];
    }
}

void dfs2(int x,int tp){
    id[x]=++cnt2;
    wt[cnt2]=w[x];
    top[x]=tp;
    if(!son[x]) return;
    dfs2(son[x],tp);
    for(int i=head[x];i;i=edge[i].nex){
        int y=edge[i].v;
        if(y==fa[x]||y==son[x]) continue;
        dfs2(y,y);
    }
}

int main(){
    scanf("%d%d%d%d",&n,&m,&r,&Mod);
    for(int i=1;i<=n;++i)
        scanf("%d",&w[i]);
    for(int i=0;i<n-1;++i){
        int x,y;
        scanf("%d%d",&x,&y);
        adde(x,y);
        adde(y,x);
    }
    dfs1(r,0,1);
    dfs2(r,r);
    build(1,1,n);
    while(m--){
        int op,x,y,z;
        scanf("%d",&op);
        if(op==1){
            scanf("%d%d%d",&x,&y,&z);
            updRange(x,y,z);
        }
        else if(op==2){
            scanf("%d%d",&x,&y);
            printf("%d\n",qRange(x,y));
        }
        else if(op==3){
            scanf("%d%d",&x,&z);
            updSon(x,z);
        }
        else{
            scanf("%d",&x);
            printf("%d\n",qSon(x));
        }
    }
    return 0;
}

标签：剖分,int,siz,tr,树链,maxn,res,id,模板
来源： https://www.cnblogs.com/FrankChen831X/p/11181530.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。