BZOJ[4034][HAOI2015]树上操作 树链剖分+线段树

题目链接http://www.lydsy.com/JudgeOnline/problem.php?id=4034

Description

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

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

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

Sample Input
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
Sample Output
6
9
13
HINT

对于 100% 的数据， N,M<=100000 ，且所有输入数据的绝对值都不会超过 10^6 。

题目大意：给出一棵树，支持单点修改，子树修改和区间求和

树上区间和，树剖一下套个线段树就可以了…

注意答案爆int

代码如下:

#include<cstdio>
#define N 100500
#define int long long
using namespace std;
int n,x,T,top,m,k,y,z;
int fir[N],dep[N],Top[N],tree[N],pre[N],son[N],size[N],fa[N],w[N];
struct Edge{
    int to,nex;
    Edge(){}
    Edge(int _to,int _nex):to(_to),nex(_nex){}
}nex[N*2];
struct Seg{
    int l,r,sum,lazy;
}a[N*8];
inline void swap(int&a,int&b){int c=a;a=b;b=c;}
inline void add(int x,int y){
    nex[++top]=Edge(y,fir[x]);
    fir[x]=top;
}
void dfs1(int x,int Dep,int Fa){
    fa[x]=Fa;dep[x]=Dep;
    size[x]=1;
    for(int i=fir[x];i;i=nex[i].nex){
        if(nex[i].to==Fa) continue;
        dfs1(nex[i].to,Dep+1,x);
        size[x]=size[x]+size[nex[i].to];
        if(size[nex[i].to]>size[son[x]]) son[x]=nex[i].to;
    }
}
void dfs2(int x,int topp){
    Top[x]=topp;
    tree[x]=++T;
    pre[T]=x;
    if(!son[x]) return;
    dfs2(son[x],topp);
    for(int i=fir[x];i;i=nex[i].nex){
        if(nex[i].to==fa[x] || nex[i].to==son[x]) continue;
        dfs2(nex[i].to,nex[i].to);
    }
}
void Pushdown(int x){
    a[2*x].lazy+=a[x].lazy;
    a[2*x+1].lazy+=a[x].lazy;
    a[2*x].sum=a[2*x].sum+(a[2*x].r-a[2*x].l+1)*a[x].lazy;
    a[2*x+1].sum=a[2*x+1].sum+(a[2*x+1].r-a[2*x+1].l+1)*a[x].lazy;
    a[x].lazy=0;
}
void maketree(int l,int r,int k){
    a[k].l=l;a[k].r=r;a[k].lazy=0;
    if(l==r){
        a[k].sum=w[pre[l]];
        return;
    }
    int mid=(l+r)>>1;
    maketree(l,mid,2*k);maketree(mid+1,r,2*k+1);
    a[k].sum=a[k*2].sum+a[k*2+1].sum;
}
int Query_Sum(int x,int y,int k){
    if(x==a[k].l && y==a[k].r){
        return a[k].sum;
    }
    Pushdown(k);
    int mid=(a[k].l+a[k].r)>>1;
    if(y<=mid){return Query_Sum(x,y,2*k);}
    if(x>mid){return Query_Sum(x,y,2*k+1);}
    return Query_Sum(x,mid,2*k)+Query_Sum(mid+1,y,2*k+1);
}
int Find_Sum(int x,int y){
    int sum=0;
    while(Top[x]!=Top[y]){
        if(dep[Top[x]]<dep[Top[y]]) swap(x,y);
        sum=sum+Query_Sum(tree[Top[x]],tree[x],1);
        x=fa[Top[x]];
    }
    if(tree[y]>tree[x]) swap(x,y);
    sum=sum+Query_Sum(tree[y],tree[x],1);
    return sum;
}
void Add(int x,int y,int val,int k){
    if(x>y) swap(x,y);
    if(a[k].r<x || a[k].l>y) return;
    if(x<=a[k].l && y>=a[k].r){
        a[k].lazy+=val;
        a[k].sum+=val*(a[k].r-a[k].l+1);
        return;
    }
    Pushdown(k);
    Add(x,y,val,2*k);
    Add(x,y,val,2*k+1);
    a[k].sum=a[2*k].sum+a[2*k+1].sum;
}
main(){
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;i++) scanf("%lld",w+i);
    for(int i=2;i<=n;i++){
        scanf("%lld%lld",&x,&y);
        add(x,y);add(y,x);
    }
    dfs1(1,1,0);dfs2(1,1);
    maketree(1,n,1);
    for(int i=1;i<=m;i++){
        scanf("%d",&k);
        switch(k){
            case 1: scanf("%lld%lld",&x,&y);Add(tree[x],tree[x],y,1);break;
            case 2: scanf("%lld%lld",&x,&y);Add(tree[x],tree[x]+size[x]-1,y,1);break;
            case 3: scanf("%lld",&x);printf("%lld\n",Find_Sum(x,1));break;
        }
    }
return 0;
}