洛谷 P3178 [HAOI2015]树上操作 树链剖分

题目描述

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个操作，分为三种：操作 1 ：把某个节点 x 的点权增加 a 。操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

输入输出格式

输入格式：
第一行包含两个整数 N, M 。表示点数和操作数。接下来一行 N 个整数，表示树中节点的初始权值。接下来 N-1 行每行两个正整数 from, to ， 表示该树中存在一条边 (from, to) 。再接下来 M 行，每行分别表示一次操作。其中第一个数表示该操作的种类（ 1-3 ） ，之后接这个操作的参数（ x 或者 x a ） 。

输出格式：
对于每个询问操作，输出该询问的答案。答案之间用换行隔开。

输入输出样例

输入样例#1：
5 5
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3
输出样例#1：
6
9
13
说明

对于 100% 的数据， N,M<=100000 ，且所有输入数据的绝对值都不

会超过 10^6 。

分析：一看就是树链剖分的模版题。tm改了5h，一些智障错误，子树区间搞错，没开longlong，线段树lazy什么的，输入会有负数……

代码：

#include<bits/stdc++.h> 
using namespace std;
#define LL long long
const int maxn=500000+10;
inline LL gi(){
    char ch=getchar();LL x=0,q=0;
    while(ch<'0' || ch>'9') ch=='-'?q=1:0,ch=getchar();
    while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return q?(-x):x;
}
#define mid ((l+r)>>1)  
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define len (r-l+1)
LL n,m;
struct Edge{
    LL next,to;

}h[maxn<<1];
LL head[maxn<<1];
LL tot,to[maxn],w[maxn],wt[maxn];
LL a[maxn<<2],laz[maxn<<2];
LL son[maxn],id[maxn],fa[maxn],cnt,dep[maxn],siz[maxn],top[maxn]; 
LL res=0;
inline void Add(LL u,LL v){
    h[++tot].next=head[u];
    h[tot].to=v;
    head[u]=tot;
}
inline void pushdown(LL rt,LL lenn){
    laz[rt<<1]+=laz[rt];
    laz[rt<<1|1]+=laz[rt];
    a[rt<<1]+=laz[rt]*(lenn-(lenn>>1));
    a[rt<<1|1]+=laz[rt]*(lenn>>1);
    laz[rt]=0;
}
inline void build(LL rt,LL l,LL r){
    if(l==r){
        a[rt]=wt[l];
        return;
    }
    build(lson);
    build(rson);
    a[rt]=(a[rt<<1]+a[rt<<1|1]);
}
inline void query(LL rt,LL l,LL r,LL L,LL R){
    if(L<=l&&r<=R){
        res+=a[rt];
        return;
    }
    else{
        if(laz[rt])pushdown(rt,len);
        if(L<=mid)query(lson,L,R);
        if(R>mid)query(rson,L,R);
    }
}
inline void update(LL rt,LL l,LL r,LL L,LL R,LL k){
    if(L<=l&&r<=R){
        laz[rt]+=k;
        a[rt]+=k*len;
    }
    else{
        if(laz[rt])pushdown(rt,len);
        if(L<=mid)update(lson,L,R,k);
        if(R>mid)update(rson,L,R,k);
        a[rt]=(a[rt<<1]+a[rt<<1|1]);
    }
}
inline LL qRange(LL x,LL y){
    LL ans=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        res=0;
        query(1,1,n,id[top[x]],id[x]);
        ans+=res; 
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    res=0;
    query(1,1,n,id[x],id[y]);
    ans+=res;
    return ans;
}
inline void updSon(LL x,LL k){
    update(1,1,n,id[x],id[x]+siz[x]-1,k);
}

inline void dfs1(LL x,LL f,LL deep){
    dep[x]=deep; 
    fa[x]=f;
    siz[x]=1; 
    LL maxson=-1;
    for(LL i=head[x];i;i=h[i].next){
        LL y=h[i].to;
        if(y==f)continue;
        dfs1(y,x,deep+1); 
        siz[x]+=siz[y]; 
        if(siz[y]>maxson)son[x]=y,maxson=siz[y];
    }
}

inline void dfs2(LL x,LL topf){
    id[x]=++cnt;
    wt[cnt]=w[x];
    top[x]=topf;
    if(!son[x])return;
    dfs2(son[x],topf);
    for(LL i=head[x];i;i=h[i].next){
        LL y=h[i].to;
        if(y==fa[x]||y==son[x])continue;
        dfs2(y,y);
    }
}

int main(){
    n=gi();m=gi();
    for(LL i=1;i<=n;i++)w[i]=gi();
    for(LL i=1;i<n;i++){
        LL a,b;
        a=gi();b=gi();
        Add(a,b);
        Add(b,a);
    }
    dfs1(1,0,1);
    dfs2(1,1);
    build(1,1,n);
    while(m--){
        LL k,x,y,z;
        k=gi();
        if(k==1){
            x=gi();y=gi();
            update(1,1,n,id[x],id[x],y);
        }
        else if(k==2){
            x=gi();y=gi();
            updSon(x,y);
        }
        else{
            x=gi();
            printf("%ld\n",qRange(1,x));
        }
    }
    return 0;
}