【题目描述】
给你一棵具有N个点(编号为1到N)M条边的树,并给定各个点权的值,然后有3种操作:
I C1 C2 K:把C1与C2的路径上的所有点权值加上K
D C1 C2 K:把C1与C2的路径上的所有点权值减去K
Q C:查询节点编号为C的权值
【输入格式】
输入包含多组测试数据。
对于每组测试数据的第一行包含三个整数N,M,P,分别表示N个结点(1<=N<=50000),M条边(M=N-1)和P(1≤P≤100000)个操作。
接下来一行为N个整数A1,A2,...AN(0≤Ai≤1000),表示一开始每个结点的值。
接下来的M行,每行两个整数u和v,表示u到v有一条边。
接下来P行,表示相应的操作,其中0<=k<=1000。
【输出格式】
对于每个询问输出一行,为询问结点的值。
树链剖分的裸题,不过对区间的操作比较简单,我们可以直接用树状数组的区间修改与单点查询,没必要写线段树,代码上会简短不少
#include<iostream>
#include<cstdio>
#include<queue>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<bitset>
using namespace std;
int n,m,q,a[50005],h[50005],cnt,size[50005],prt[50005],son[50005],deep[50005],seg[50005],top[50005],rev[50005],tree[50005];
struct edge
{
int to,next;
}w[100005];
void add(int x,int y)
{
cnt++;
w[cnt].to=y;
w[cnt].next=h[x];
h[x]=cnt;
}
void dfs(int x,int fa)
{
prt[x]=fa;
size[x]=1;
deep[x]=deep[fa]+1;
for(int i=h[x];i;i=w[i].next)
{
int to=w[i].to;
if(to==fa) continue;
dfs(to,x);
size[x]+=size[to];
if(size[to]>size[son[x]]) son[x]=to;
}
}
void dfs2(int x)
{
if(son[x])
{
top[son[x]]=top[x];
seg[son[x]]=++seg[0];
rev[seg[0]]=son[x];
dfs2(son[x]);
}
for(int i=h[x];i;i=w[i].next)
{
int to=w[i].to;
if(top[to]) continue;
top[to]=to;
seg[to]=++seg[0];
rev[seg[0]]=to;
dfs2(to);
}
}
void add2(int x,int k)
{
// cout<<x<<endl;
while(x<=n)
{
tree[x]+=k;
x+=(x&-x);
}
}
int sum(int x)
{
int ans=0;
while(x)
{
ans+=tree[x];
x-=(x&-x);
}
return ans;
}
void uprange(int x,int y,int z)
{
// cout<<x<<" "<<y<<" "<<z<<endl;
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]]) swap(x,y);
add2(seg[top[x]],z);
add2(seg[x]+1,-z);
x=prt[top[x]];
}
if(deep[x]>deep[y]) swap(x,y);
add2(seg[x],z);
add2(seg[y]+1,-z);
}
int main()
{
while(~scanf("%d%d%d",&n,&m,&q))
{
memset(tree,0,sizeof(tree));
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
add(x,y);
add(y,x);
}
dfs(1,0);
seg[0]=seg[1]=rev[1]=top[1]=1;
dfs2(1);
for(int i=1;i<=n;i++) add2(seg[i],a[i]),add2(seg[i]+1,-a[i]);
for(int i=1;i<=q;i++)
{
char s[10];
scanf("%s",s);
if(s[0]=='Q')
{
int x;
scanf("%d",&x);
printf("%d\n",sum(seg[x]));
}
else if(s[0]=='D')
{
int l,r,z;
scanf("%d%d%d",&l,&r,&z);
uprange(l,r,-z);
}
else
{
int l,r,z;
scanf("%d%d%d",&l,&r,&z);
uprange(l,r,z);
}
}
}
}