HDU 3966 Aragorn's Story（树链剖分+线段树）

Description
给一棵树，并给定各个点权的值，然后有3种操作
I C1 C2 K: 把C1与C2的路径上的所有点权值加上K
D C1 C2 K：把C1与C2的路径上的所有点权值减去K
Q C：查询节点编号为C的权值
Input
第一行为三个整数n,m和q分别表示点数，边数以及操作数，第二行n个整数表示n个节点的权值，之后m行每行两个整数a和b表示a和b之间有一条无向边，最后q行每行一种操作
Output
对于每次查询，输出查询结果
Sample Input
3 2 5
1 2 3
2 1
2 3
I 1 3 5
Q 2
D 1 2 2
Q 1
Q 3
Sample Output
7
4
8
Solution
树上路径问题，首先树链剖分，然后建线段树，树的元素存储区间权值和，问题转化为线段树单点查询与区间修改操作
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 55555
struct Edge
{
    int to,next;
}E[2*maxn];
struct Tree
{
    int left,right,data;
}T[4*maxn];
int n,m,q,val[maxn],w[maxn],head[maxn],cnt,idx,size[maxn],fa[maxn],son[maxn],dep[maxn],top[maxn],id[maxn];
void init()
{
    cnt=idx=0;
    memset(head,-1,sizeof(head));
    dep[1]=fa[1]=size[0]=0;
    memset(son,0,sizeof(son));
}
void add(int u,int v)
{
    E[cnt].to=v;
    E[cnt].next=head[u];
    head[u]=cnt++;
}
void dfs1(int u)
{
    size[u]=1;
    for(int i=head[u];~i;i=E[i].next)
    {
        int v=E[i].to;
        if(v!=fa[u])
        {
            fa[v]=u;
            dep[v]=dep[u]+1;
            dfs1(v);
            size[u]+=size[v];
            if(size[son[u]]<size[v]) son[u]=v;
        }
    }
}
void dfs2(int u,int topu)
{
    top[u]=topu;
    id[u]=++idx;
    if(son[u]) dfs2(son[u],top[u]);
    for(int i=head[u];~i;i=E[i].next)
    {
        int v=E[i].to;
        if(v!=fa[u]&&v!=son[u]) dfs2(v,v);
    }
}
void build(int l,int r,int t)
{
    T[t].left=l;
    T[t].right=r;
    T[t].data=0;
    if(l==r)
    {
        T[t].data=w[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(l,mid,2*t);
    build(mid+1,r,2*t+1);
}
void lazy(int t)
{
    if(T[t].data)
    {
        T[2*t].data+=T[t].data;
        T[2*t+1].data+=T[t].data;
        T[t].data=0;
    }
}
int query(int x,int t)
{
    if(T[t].left==x&&T[t].right==x) return T[t].data;
    lazy(t);
    if(x<=T[2*t].right) return query(x,2*t);
    return query(x,2*t+1);
}
void modify(int l,int r,int t,int z)
{
    if(T[t].left==l&&T[t].right==r)
    {
        T[t].data+=z;
        return ;
    }
    lazy(t);
    if(r<=T[2*t].right) modify(l,r,2*t,z);
    else if(l>=T[2*t+1].left) modify(l,r,2*t+1,z);
    else
    {
        modify(l,T[2*t].right,2*t,z);
        modify(T[2*t+1].left,r,2*t+1,z);
    }
}
void change(int u,int v,int z)
{
    int top1=top[u],top2=top[v];
    while(top1!=top2)
    {
        if(dep[top1]<dep[top2])
        {
            swap(top1,top2);
            swap(u,v);
        }
        modify(id[top1],id[u],1,z);
        u=fa[top1];
        top1=top[u];
    }
    if(dep[u]>dep[v]) swap(u,v);
    modify(id[u],id[v],1,z);
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&q))
    {
        init();
        for(int i=1;i<=n;i++) scanf("%d",&val[i]);
        int u,v,z;
        while(m--)
        {
            scanf("%d%d",&u,&v);
            add(u,v);add(v,u);
        }
        dfs1(1);
        dfs2(1,1);
        for(int i=1;i<=n;i++)
            w[id[i]]=val[i];
        build(1,idx,1);
        char op[11];
        while(q--)
        {
            scanf("%s",op);
            if(op[0]=='Q')
            {
                scanf("%d",&u);
                printf("%d\n",query(id[u],1));
            }
            else
            {
                scanf("%d%d%d",&u,&v,&z);
                if(op[0]=='D')z=-z;
                change(u,v,z);
            }
        }
    }
    return 0;
}