HDU 3966 Aragorn's Story(树链剖分，区间修改，单点查询)

区间修改，单点查询，很容易想到线段树，树形结构的线段树可以用树链剖分。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
#define maxn 50005
int val_pre[maxn],val[maxn],siz[maxn],son[maxn],id[maxn],fa[maxn];
int top[maxn],dep[maxn];
int topw,m,n,p;
vector<int> G[maxn];
void dfs_1(int u,int f,int d)
{
siz[u]=1; //当前节点的子节点数目初始化为1
fa[u]=f; //父节点
dep[u]=d; //当前节点的深度
son[u]=0; //最重的儿子
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
if(v==f) continue;
dfs_1(v,u,d+1);
siz[u]+=siz[v];
if(siz[son[u]]<siz[v])son[u]=v;
}
}
void dfs_2(int u,int tp)
{
top[u]=tp; //当前节点的所在重链的链手编号
id[u]=++topw; //重新编号
if(son[u]) dfs_2(son[u],tp);
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
if(v==fa[u]||v==son[u])continue;
dfs_2(v,v);
}
}

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int add[maxn<<2];

void push_down(int rt) //更新lazy标记
{
if(add[rt])
{
add[rt<<1]+=add[rt];
add[rt<<1|1]+=add[rt];
add[rt]=0;
}
}
void build(int l,int r,int rt) //建树
{
add[rt]=0;
if(l==r)
{
add[rt]=val[r];
return ;
}
int m=(l+r)>>1;
build(lson);
build(rson);
}
void update(int L,int R,int c,int l,int r,int rt)
{
if(L<=l&&R>=r)
{
add[rt]+=c;
return ;
}
push_down(rt);
int m=(l+r)>>1;
if(L<=m)update(L,R,c,lson);
if(R>m)update(L,R,c,rson);
}
int query(int x,int l,int r,int rt)
{
if(l==r)
return add[rt];
push_down(rt); //查找过程中要更新lazy标记
int m=(l+r)>>1;
if(x<=m)query(x,lson);
else query(x,rson);
}
void change(int u,int v,int add)
{
int fu=top[u],fv=top[v];
while(fu!=fv) //往上一直找到在同一条重链中
{
if(dep[fu]<dep[fv])
{
swap(u,v);swap(fu,fv);
}
update(id[fu],id[u],add,1,topw,1);//更新lazy标记
u=fa[fu];
fu=top[u];
}
if(u==v)
{
update(id[u],id[v],add,1,topw,1);
}
else
{
if(id[u]>id[v]) swap(u,v); //编号小的在前面
update(id[u],id[v],add,1,topw,1);
return ;
}
}
int main()
{
while(~scanf("%d%d%d",&n,&m,&p))
{
for(int i=1;i<=n;i++)
{
scanf("%d",&val_pre[i]);
}
for(int i=1,a,b;i<=m;i++)
{
scanf("%d%d",&a,&b);
G[a].push_back(b);
G[b].push_back(a);
}
topw=0;
dfs_1(1,0,1);
dfs_2(1,1);
for(int i=1;i<=n;i++)
{
val[id[i]]=val_pre[i];
}
build(1,topw,1);
char op[10];
int a,b,c;
while(p--)
{
scanf("%s",op);
if(op[0]=='I')
{
scanf("%d%d%d",&a,&b,&c);
change(a,b,c);
}
else if(op[0]=='D')
{
scanf("%d%d%d",&a,&b,&c);
change(a,b,-c);
}
else
{
scanf("%d",&a);
int ans = query(id[a],1,topw,1);
printf("%d\n",ans);
}
}
for(int i=0;i<maxn;i++) G[i].clear();
}

return 0;
}