BZOJ4034：[HAOI2015]树上操作——题解

http://www.lydsy.com/JudgeOnline/problem.php?id=4034

https://www.luogu.org/problemnew/show/P3178

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个

操作，分为三种：

操作 1 ：把某个节点 x 的点权增加 a 。

操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。

操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

树剖板子。不会写树链剖分的请看：http://www.cnblogs.com/luyouqi233/p/7886709.html

试图默写结果被1和l搞反然后gg。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int N=1e5+;
inline int read(){
    int X=,w=;char ch=;
    while(ch<''||ch>''){w|=ch=='-';ch=getchar();}
    while(ch>=''&&ch<='')X=(X<<)+(X<<)+(ch^),ch=getchar();
    return w?-X:X;
}
struct node{
    int to,nxt;
}edge[*N];
int head[N],cnt,tot,n,q;
inline void add(int u,int v){
    edge[++cnt].to=v;edge[cnt].nxt=head[u];head[u]=cnt;
}
int fa[N],son[N],size[N],val[N],dep[N],idx[N],pos[N],top[N];
ll sum[*N],lazy[*N];
void dfs1(int u){
    size[u]=;
    for(int i=head[u];i;i=edge[i].nxt){
    int v=edge[i].to;
    if(v==fa[u])continue;
    fa[v]=u;dep[v]=dep[u]+;
    dfs1(v);
    size[u]+=size[v];
    if(!son[u]||size[son[u]]<size[v])son[u]=v;
    }
}
void dfs2(int u,int anc){
    pos[u]=++tot;
    idx[tot]=u;
    top[u]=anc;
    if(!son[u])return;
    dfs2(son[u],anc);
    for(int i=head[u];i;i=edge[i].nxt){
    int v=edge[i].to;
    if(v==fa[u]||v==son[u])continue;
    dfs2(v,v);
    }
}
void init(){
    dep[]=;
    dfs1();
    dfs2(,);
}
void push(int a,int l,int r){
    int mid=(l+r)>>;
    if(lazy[a]){
    lazy[a*]+=lazy[a];
    lazy[a*+]+=lazy[a];
    sum[a*]+=(mid-l+)*lazy[a];
    sum[a*+]+=(r-mid)*lazy[a];
    lazy[a]=;
    }
}
void build(int a,int l,int r){
    if(l==r){
    sum[a]=val[idx[l]];
    return;
    }
    int mid=(l+r)>>;
    build(a*,l,mid);
    build(a*+,mid+,r);
    sum[a]=sum[a*]+sum[a*+];
}
ll query(int a,int l,int r,int l1,int r1){
    if(r1<l||r<l1)return ;
    if(l1<=l&&r<=r1)return sum[a];
    int mid=(l+r)>>;
    push(a,l,r);
    return query(a*,l,mid,l1,r1)+query(a*+,mid+,r,l1,r1);
}
ll pathquery(int x,int y){
    ll res=;
    while(top[x]!=top[y]){
    if(dep[top[x]]<dep[top[y]])swap(x,y);
    res+=query(,,n,pos[top[x]],pos[x]);
    x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    res+=query(,,n,pos[x],pos[y]);
    return res;
}
void modify(int a,int l,int r,int l1,int r1,ll p){
    if(r1<l||r<l1)return;
    if(l1<=l&&r<=r1){
    sum[a]+=(r-l+)*p;
    lazy[a]+=p;
    return;
    }
    int mid=(l+r)>>;
    push(a,l,r);
    modify(a*,l,mid,l1,r1,p);
    modify(a*+,mid+,r,l1,r1,p);
    sum[a]=sum[a*]+sum[a*+];
}
int main(){
    n=read(),q=read();
    for(int i=;i<=n;i++)val[i]=read();
    for(int i=;i<n;i++){
    int u=read(),v=read();
    add(u,v);add(v,u);
    }
    init();
    build(,,n);
    while(q--){
    int op=read();
    if(op==){
        int x=read(),a=read();
        modify(,,n,pos[x],pos[x],a);
    }
    if(op==){
        int x=read(),a=read();
        modify(,,n,pos[x],pos[x]+size[x]-,a);
    }
    if(op==){
        int x=read();
        printf("%lld\n",pathquery(x,));
    }
    }
    return ;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。 　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++