[BZOJ]4034: [HAOI2015]树上操作

[HAOI2015]树上操作

传送门

题目大意：三个操作 1：a,b,c b节点权值+c 2：a,b,c 以b为根的子树节点权值全部+c 3:a,b 查询b到根路径的权值和。

题解：树链剖分

操作1 ,2是区间修改，3是区间和。

看题解都提示开long long 了，我也开了，可是整形相乘赋值给Long long时爆了。

简直要哭晕了。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 500005
#define LL long long
using namespace std;

int sumedge,cnt,n,m;
int head[maxn],size[maxn],dad[maxn],deep[maxn],top[maxn];
int tpos[maxn],re[maxn],w[maxn];

struct Tree{
    int l,r;LL sum,s;
}tr[maxn<<];

struct Edge{
    int x,y,nxt;
    Edge(int x=,int y=,int nxt=):
        x(x),y(y),nxt(nxt){}
}edge[maxn<<];

void read(int &x){
    char ch=getchar();x=;int f=;
    for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-;
    for(;isdigit(ch);ch=getchar())x=x*+ch-'';
    x=x*f;
}

void add(int x,int y){
    edge[++sumedge]=Edge(x,y,head[x]);
    head[x]=sumedge;
}

void dfs(int x){
    size[x]=;deep[x]=deep[dad[x]]+;
    for(int i=head[x];i;i=edge[i].nxt){
        int v=edge[i].y;
        if(v==dad[x])continue;
        dad[v]=x;dfs(v);
        size[x]+=size[v];
    }
}

void dfs_(int x){
    int s=;tpos[x]=++cnt;re[cnt]=x;
    if(!top[x])top[x]=x;
    for(int i=head[x];i;i=edge[i].nxt){
        int v=edge[i].y;
        if(v!=dad[x]&&size[v]>size[s])s=v;
    }
    if(s){
        top[s]=top[x];
        dfs_(s);
    }
    for(int i=head[x];i;i=edge[i].nxt){
        int v=edge[i].y;
        if(v!=dad[x]&&v!=s)dfs_(v);
    }
}

void pushup(int rt){
    tr[rt].sum=tr[rt<<].sum+tr[rt<<|].sum;
    return;
}

void pushdown(int rt){
    if(!tr[rt].s)return;
    tr[rt<<].s+=tr[rt].s;tr[rt<<|].s+=tr[rt].s;
    tr[rt<<].sum+=(LL)(tr[rt<<].r-tr[rt<<].l+)*tr[rt].s;
    tr[rt<<|].sum+=(LL)(tr[rt<<|].r-tr[rt<<|].l+)*tr[rt].s;
    tr[rt].s=;return;
}

void build(int rt,int l,int r){
    tr[rt].l=l;tr[rt].r=r;
    if(l==r){
        tr[rt].sum=w[re[l]];
        return;
    }
    int mid=(l+r)>>;
    build(rt<<,l,mid);build(rt<<|,mid+,r);
    pushup(rt);
}

void change(int rt,int l,int r,int ql,int qr,int p){
    if(l>=ql&&r<=qr){
        tr[rt].sum+=(LL)(r-l+)*p;
        tr[rt].s+=p;
        return;
    }
    pushdown(rt);
    int mid=(l+r)>>;
    if(ql<=mid)change(rt<<,l,mid,ql,qr,p);
    if(qr>mid)change(rt<<|,mid+,r,ql,qr,p);
    pushup(rt);
}

LL query_sum(int rt,int l,int r,int ql,int qr){
    if(l>=ql&&r<=qr){
        return tr[rt].sum;
    }
    pushdown(rt);
    int mid=(l+r)>>;LL ans=;
    if(ql<=mid)ans+=query_sum(rt<<,l,mid,ql,qr);
    if(qr>mid)ans+=query_sum(rt<<|,mid+,r,ql,qr);
    return ans;
}

LL query(int x){
    LL ret=;
    for(;top[x]!=;){
        ret+=query_sum(,,n,tpos[top[x]],tpos[x]);
        x=dad[top[x]];
    }
    ret+=query_sum(,,n,,tpos[x]);
    return ret;
}

int main(){
    read(n);read(m);
    for(int i=;i<=n;i++)read(w[i]);
    for(int i=;i<n;i++){
        int x,y;
        read(x);read(y);
        add(x,y);add(y,x);
    }
    dfs();dfs_();build(,,n);
    for(int i=;i<=m;i++){
        int od,x,a;
        read(od);read(x);
        if(od==){
            read(a);
            change(,,n,tpos[x],tpos[x],a);
        }else if(od==){
            read(a);
            change(,,n,tpos[x],tpos[x]+size[x]-,a);
        }else if(od==){
            printf("%lld\n",query(x));
        }
    }
    return ;
}

AC