树上区间更新和查询,原理就是把树拆成几条链,dfs序是连续的,所以就把他们强加到线段树上,进行更新和查询。
进行两遍dfs,第一遍把深度,重儿子,父节点等一些信息找出,然后第二遍dfs时记录每个节点的链头,遍历时先走重儿子,把树上的每个节点按dfs序应到数组上,建树就行了。具体过程找个视频看看就懂了。写的不太规范,自己能看懂就好了。
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
struct code
{
int l,r,lazy;
} tree[50005*4];
vector<int>dap[50005];
int id_sum[50005],s,e,d;
int n,m,q,siz[50005],deep[50005],son[50005];
int sum[50005],top[50005],fa[50005],tri[50005];
void buid(int l,int r,int t)
{
tree[t].l=l;
tree[t].r=r;
tree[t].lazy=0;
if(l==r)
{
tree[t].lazy=id_sum[l];
return ;
}
int mid=(l+r)/2;
buid(l,mid,2*t);
buid(mid+1,r,2*t+1);
}
void update(int l,int r,int t,int d)
{
if(tree[t].l==l&&tree[t].r==r)
{
tree[t].lazy+=d;
return ;
}
if(tree[t].lazy!=0)
{
tree[2*t].lazy+=tree[t].lazy;
tree[2*t+1].lazy+=tree[t].lazy;
tree[t].lazy=0;
}
int mid=(tree[t].l+tree[t].r)/2;
if(r<=mid)
update(l,r,2*t,d);
else if(l>mid)
update(l,r,2*t+1,d);
else
{
update(l,mid,2*t,d);
update(mid+1,r,2*t+1,d);
}
}
int queuy(int t,int p)
{
if(tree[t].l==tree[t].r)
return tree[t].lazy;
if(tree[t].lazy!=0)
{
tree[2*t].lazy+=tree[t].lazy;
tree[2*t+1].lazy+=tree[t].lazy;
tree[t].lazy=0;
}
int mid=(tree[t].l+tree[t].r)/2;
if(p<=mid)
return queuy(2*t,p);
else
return queuy(2*t+1,p);
}
void dfs(int x,int from)
{
deep[x]=deep[from]+1;
siz[x]=1;
fa[x]=from;
son[x]=0;
int len=dap[x].size();
for(int i=0; i<len; i++)
{
int ty=dap[x][i];
if(ty==from)
continue;
dfs(ty,x);
siz[x]+=siz[ty];
if(siz[ty]>siz[son[x]])
son[x]=ty;
}
}
int cnt;
void qdfs(int x,int t)
{
top[x]=t,tri[x]=cnt++;
id_sum[cnt-1]=sum[x];
if(son[x])qdfs(son[x],t);
int len=dap[x].size();
for(int i=0; i<len; i++)
{
int ty=dap[x][i];
if(ty==son[x]||ty==fa[x])
continue;
qdfs(ty,ty);
}
}
void init()
{
cnt=1;
deep[1]=0;
dfs(1,1);
qdfs(1,1);
buid(1,n,1);
}
void Update(int s,int e,int d)
{
int tx=top[s],ty=top[e];
while(tx!=ty)
{
if(deep[tx]<deep[ty])
swap(s,e),swap(tx,ty);
update(tri[tx],tri[s],1,d);
s=fa[tx],tx=top[s];
}
if(deep[s]<deep[e])
swap(s,e);
update(tri[e],tri[s],1,d);
}
char str[5];
int main()
{
while(~scanf("%d%d%d",&n,&m,&q))
{
for(int i=1; i<=n; i++)
{
scanf("%d",&sum[i]);
dap[i].clear();
}
for(int i=1; i<=m; i++)
{
scanf("%d%d",&s,&e);
dap[s].push_back(e);
dap[e].push_back(s);
}
init();
for(int i=0;i<q;i++)
{
scanf("%s",str);
if(str[0]=='Q')
{
scanf("%d",&s);
printf("%d\n",queuy(1,tri[s]));
}
else
{
scanf("%d%d%d",&s,&e,&d);
if(str[0]=='I')
Update(s,e,d);
else
Update(s,e,-d);
}
}
}
return 0;
}