HDU - 3966 Aragorn's Story 树链剖分 基于点+树状数组 模板

题目链接：点击查看

题意：给出n个点的树，和每个点权值，三种操作，查询点的权值，增加、减少路径上每个点的权值

题解：树链剖分模板题；讲解：点击查看

#include <bits/stdc++.h>
using namespace std;
const int N=5e4+10;
#define lowbit(x) x&(-x)
struct node
{
    int to,nex;
}e[N * 2];
struct point
{
    int val,top,fa,deep,num,son;
}a[N];
int head[N],len;
int id[N],pid[N],cnt;
int sum[N];
int n,m,p;
void init()
{
    for(int i=1;i<=n;i++)
    {
        head[i]=-1;
        a[i].son=-1;
        sum[i]=0;
    }
    cnt=0;
    len=0;
}
void add(int u,int v)
{
    e[len].to=v;
    e[len].nex=head[u];
    head[u]=len++;
}
void dfs(int u,int fa,int dep)
{
    a[u].deep=dep;
    a[u].fa=fa;
    a[u].num=1;
    int to;
    for(int i=head[u];i!=-1;i=e[i].nex)
    {
        to=e[i].to;
        if(to==fa) continue;
        dfs(to,u,dep+1);
        a[u].num+=a[to].num;
        if(a[u].son==-1||a[to].num>a[a[u].son].num)
        {
            a[u].son=to;
        }
    }
}
void update(int x,int val)
{
    while(x<=n)
    {
        sum[x]+=val;
        x+=lowbit(x);
    }
}
int query(int x)
{
    int res=0;
    while(x)
    {
        res+=sum[x];
        x-=lowbit(x);
    }
    return res;
}
void getpos(int u,int topfa)
{
    a[u].top=topfa;
    id[u]=++cnt;
    pid[cnt]=u;
    if(a[u].son==-1) return;
    getpos(a[u].son,topfa);
    int to;
    for(int i=head[u];i!=-1;i=e[i].nex)
    {
        to=e[i].to;
        if(to==a[u].fa || to==a[u].son) continue;
        getpos(to,to);
    }
}
void change(int u,int v,int val)
{
    int f1=a[u].top,f2=a[v].top;
    while(f1!=f2)
    {
        if(a[f1].deep<a[f2].deep)
        {
            swap(f1,f2);
            swap(u,v);
        }
        update(id[f1],val);
        update(id[u]+1,-val);
        u=a[f1].fa;
        f1=a[u].top;
    }
    if(a[u].deep>a[v].deep) swap(u,v);
    update(id[u],val);
    update(id[v]+1,-val);
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&p))
    {
        init();
        int u,v,k;
        char op[3];
        for(int i=1;i<=n;i++) scanf("%d",&a[i].val);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
        }
        dfs(1,0,0);
        getpos(1,1);
        for(int i=1;i<=n;i++)
        {
            update(id[i],a[i].val);
            update(id[i]+1,-a[i].val);
        }
        while(p--)
        {
            scanf("%s",op);
            if(op[0]=='Q')
            {
                scanf("%d",&u);
                printf("%d\n",query(id[u]));
            }
            else
            {
                scanf("%d%d%d",&u,&v,&k);
                if(op[0]=='D') k=-k;
                change(u,v,k);
            }
        }
    }
    return 0;
}