题目链接:点击查看
题意:给出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;
}