HDU 3966 Aragorn's Story(树链剖分 点权，树状数组)

题目链接
题意：给出一棵树，每次把路径(u,v)上的点权都加上或者减去一个值，然后问某一点的值。 树链剖分，后面可以使用线段树，也可以使用树状数组。使用树状数组的话，因为是点权，要移到边权上，所以pos初始为1。

树状数组，使用了区间的更新，单点查询。
具体的过程是初始化 add(i,x),add(i+1,-x)
这样sum(x)就是x点的值了。而不会是x的前缀和了

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<stack>
#include<cstring>
#include<set>
#include<map>
#include<string>
#include<cassert>
using namespace std;
#define cl(a,b) memset(a,b,sizeof(a))
#define fastIO ios::sync_with_stdio(false);cin.tie(0);
#define ll unsigned long long
#define pb push_back
#define gcd __gcd


#define For(i,j,k) for(int i=(j);i<=k;i++)
#define lowbit(i) (i&(-i))
#define _(x) printf("%d\n",x)

typedef vector<ll> vec;
typedef pair<int,int> PI;
const double EPS = 1e-8;
const int maxn = 5e4+100;
const int inf  = 1 << 28;

int val[maxn];
struct Edge{
    int to,next;
}es[maxn<<1];

int head[maxn],tot;

int top[maxn];
int fa[maxn];
int p[maxn];
int fp[maxn];
int num[maxn];
int son[maxn];
int deep[maxn];
int pos;

void addEdge(int u,int v){
    es[tot].to=v;es[tot].next=head[u];head[u]=tot++;
}
void init(){
    tot=0;pos=1;
    memset(head,-1,sizeof(head));
    memset(son,-1,sizeof(son));
}

void dfs(int u,int pre,int d){
    num[u]=1;
    deep[u]=d;
    fa[u]=pre;
    for(int i=head[u];~i;i=es[i].next){
        int v=es[i].to;
        if(v!=pre){
            dfs(v,u,d+1);
            num[u]+=num[v];
            if(son[u]==-1||num[son[u]]<num[v])son[u]=v;
        }
    }
}
void getpos(int u,int sp){
    top[u]=sp;
    p[u]=pos++;
    fp[p[u]]=u;
    if(son[u]==-1)return;
    getpos(son[u],sp);
    for(int i=head[u];~i;i=es[i].next){
        int v=es[i].to;
        if(v!=fa[u]&&v!=son[u])getpos(v,v);
    }
}

int n,m,q;
int c[maxn];
void add(int i,int v){
    while(i<=n){
        c[i]+=v;
        i+=lowbit(i);
    }
}
int sum(int i){
    int s=0;
    while(i>0){
        s+=c[i];i-=lowbit(i);
    }
    return s;
}

void change(int u,int v,int val){
    int f1=top[u],f2=top[v];
    while(f1!=f2){
        if(deep[f1]<deep[f2]){
            swap(f1,f2);
            swap(u,v);
        }
        add(p[f1],val);
        add(p[u]+1,-val);
        u=fa[f1];f1=top[u];
    }
    if(deep[u]>deep[v])swap(u,v);
    add(p[u],val);
    add(p[v]+1,-val);
}

int main(){
    while(~scanf("%d%d%d",&n,&m,&q)){
        init();
        for(int i=1;i<=n;i++){
            scanf("%d",&val[i]);
        }
        for(int i=0;i<n-1;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            addEdge(x,y);addEdge(y,x);
        }
        dfs(1,0,0);
        getpos(1,1);
        memset(c,0,sizeof(c));
        for(int i=1;i<=n;i++){
            add(p[i],val[i]);
            add(p[i]+1,-val[i]);;
        }
        char op[23];
        while(q--){
            scanf("%s",op);
            if(op[0]=='I'){
                int x,y,z;
                scanf("%d%d%d",&x,&y,&z);
                change(x,y,z);
            }
            else if(op[0]=='D'){
                int x,y,z;
                scanf("%d%d%d",&x,&y,&z);
                change(x,y,-z);
            }
            else {
                int x;scanf("%d",&x);
                printf("%d\n",sum(p[x]));
            }
        }
    }
    return 0;
}