标签：ll 剖分 int top 树链 ans son id 模板

#include<bits/stdc++.h>
#define lson(x) (x << 1)
#define rson(x) (x << 1 | 1)
using namespace std;

const int N = 100010,M = 200010;
int n,m,r,p;
long long a[N];
int d[N],son[N],fa[N],siz[N],top[N],rk[N],id[N],cnt;//树链剖分 
int head[N],edge[M],nxt[M],from[M],to[M],tot;//邻接表 
void add_edge(int x,int y){
    from[++tot] = x;
    to[tot] = y;
    nxt[tot] = head[x];
    head[x] = tot;
}

//线段树
struct Tree{
    int l,r;
    long long sum,tag;  
}t[N*4];
void push_up(int x) { t[x].sum = (t[lson(x)].sum + t[rson(x)].sum) % p; }
void push_down(int x){
    if(t[x].tag){
        t[lson(x)].tag = (t[lson(x)].tag + t[x].tag) % p; 
        t[lson(x)].sum = (t[lson(x)].sum + t[x].tag *(t[lson(x)].r - t[lson(x)].l + 1)) % p;
        t[rson(x)].tag = (t[rson(x)].tag + t[x].tag) % p; 
        t[rson(x)].sum = (t[rson(x)].sum + t[x].tag *(t[rson(x)].r - t[rson(x)].l + 1)) % p;
        t[x].tag = 0;
    }
}
void build(int x,int ll,int rr){
    t[x].l = ll; t[x].r = rr;
    if(ll == rr) { 
        t[x].sum = a[rk[ll]]; 
        if(t[x].sum > p) t[x].sum %= p; 
        return; 
    }
    int mid = (ll + rr) >> 1;
    build(lson(x),ll,mid);
    build(rson(x),mid + 1,rr);
    push_up(x);
}
void update(int x,int ll,int rr,int k){
    if(ll <= t[x].l && t[x].r <= rr){
        t[x].sum = (t[x].sum + k * (t[x].r - t[x].l + 1)) % p;
        t[x].tag = (t[x].tag + k) % p;
        return;
    }
    push_down(x);
    int mid = (t[x].l + t[x].r) >> 1;
    if(ll <= mid) update(lson(x),ll,rr,k);
    if(mid < rr)  update(rson(x),ll,rr,k);
    push_up(x);
}
long long query(int x,int ll,int rr){
    if(ll <= t[x].l && t[x].r <= rr) return t[x].sum;
    int mid = (t[x].l + t[x].r) >> 1;
    long long res = 0;
    push_down(x);
    if(ll <= mid) res = (res + query(lson(x),ll,rr)) % p;
    if(mid < rr)  res = (res + query(rson(x),ll,rr)) % p;
    push_up(x);
    return res;
}

//树链剖分 
void dfs1(int x){
    siz[x] = 1;
    for(int i = head[x]; i; i = nxt[i])
        if(!siz[to[i]]){
            d[to[i]] = d[x] + 1;
            fa[to[i]] = x;
            dfs1(to[i]);
            siz[x] +=  siz[to[i]];
            if(!son[x] || siz[to[i]] > siz[son[x]])
                son[x] = to[i];
        }
}
void dfs2(int x,int t){
    top[x] = t;
    id[x] = ++cnt;
    rk[cnt] = x;
    if(!son[x]) return;
    dfs2(son[x],t);
    for(int i = head[x]; i; i = nxt[i])
        if(to[i] != son[x] && to[i] != fa[x])
            dfs2(to[i],to[i]);
}
int main(){
//    freopen("testdata (6).in","r",stdin);
//    freopen("test.txt","w",stdout);
    cin >> n >> m >> r >> p;
    for(int i = 1; i <= n; i++)
        scanf("%lld",&a[i]);
    for(int i = 1; i < n; i++){
        int x,y;
        scanf("%d%d",&x,&y);
        add_edge(x,y);
        add_edge(y,x);
    }
    dfs1(r);
    dfs2(r,r);
    build(1,1,n);
    for(int i = 1; i <= m; i++){
        int op = 0,x = 0,y = 0,z = 0;
        scanf("%d%d",&op,&x);
        if(op == 1){
            scanf("%d%d",&y,&z);
            while(top[x] != top[y]){
                if(d[top[y]] > d[top[x]]) swap(x,y);
                update(1,id[top[x]],id[x],z);
                x = fa[top[x]];
            }
            if(d[x] > d[y]) swap(x,y);
                update(1,id[x],id[y],z);
        }
        if(op == 2){
            scanf("%d",&y);
            long long ans = 0;
            while(top[x] != top[y]){
                if(d[top[y]] > d[top[x]]) swap(x,y);
                ans = (ans + query(1,id[top[x]],id[x])) % p;
                x = fa[top[x]];
            }
            if(d[x] > d[y]) swap(x,y);
            ans = (ans + query(1,id[x],id[y]))% p;
            printf("%lld\n",ans);
        }
        if(op == 3){
            scanf("%d",&z);
            update(1,id[x],id[x] + siz[x] - 1,z);
        }
        if(op == 4){
            printf("%lld\n",query(1,id[x],id[x] + siz[x] - 1));
        } 
    }
    return 0;
}

标签：ll,剖分,int,top,树链,ans,son,id,模板
来源： https://www.cnblogs.com/FoxC/p/11256624.html

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