【题目背景】
那个很神奇的诗歌鉴赏树上每一个知识点都有一个重要度,但是捏,新课标那个不和谐的东西决定改变语文考试的侧重点。
新课标一共有q次操作,分别为以下两种:
1. 将以知识点i为根的子树除i之外的点的重要度增加delta
2. 询问当前点重要度。
【输入格式】
第一行,知识点数n,操作数q.
第二行,第一个知识点的重要度(这个知识点是根)
以后n-1行,每行2个数,分别为该知识点的重要度和该知识点的父亲
以后m行,为一个字符和若干个整数,若字符为p,紧接着有两个整数,整数i和整数delta,表示将以知识点i为根的子树除i之外的点的重要度增加delta;若字符为u,紧接着为一个整数x,表示询问当前点x的重要度。
【输出格式】
对于每个询问,输出重要度。
树链剖分的水题,通过size直接实现对子树的操作,因为只需要区间修改和单点查询,所以我们使用较为简单的树状数组
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
int n,m,v[500005],h[500005],cnt,deep[500005],size[500005],top[500005],seg[500005],prt[500005],son[500005],tree[500005];
struct edge
{
int to,next;
}w[500005];
void add(int x,int y)
{
cnt++;
w[cnt].to=y;
w[cnt].next=h[x];
h[x]=cnt;
}
void dfs(int x)
{
size[x]=1;
for(int i=h[x];i;i=w[i].next)
{
int to=w[i].to;
deep[to]=deep[x]+1;
prt[to]=x;
dfs(to);
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];
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];
dfs2(to);
}
}
void addsum(int x,int k)
{
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;
}
int main()
{
scanf("%d%d",&n,&m);
scanf("%d",&v[1]);
for(int i=2;i<=n;i++)
{
scanf("%d",&v[i]);
int x;
scanf("%d",&x);
add(x,i);
}
deep[1]=1;
dfs(1);
seg[0]=seg[1]=top[1]=1;
dfs2(1);
for(int i=1;i<=n;i++) addsum(seg[i],v[i]),addsum(seg[i]+1,-v[i]);
for(int i=1;i<=m;i++)
{
char s[10];
scanf("%s",s);
if(s[0]=='u')
{
int x;
scanf("%d",&x);
printf("%d\n",sum(seg[x]));
}
else
{
int x,y;
scanf("%d%d",&x,&y);
addsum(seg[x]+1,y);
addsum(seg[x]+size[x],-y);
}
}
}