题意: 两种操作,1.给出一个节点u,u的权值加上val,u的下一层孩子权值加上-val,下下层孩子加上val,一直
2.查询节点u的权值
思路:将树进行离散化,跑个dfs,然后节点和它的孩子节点都在一个连续的区间内。
更新的时候,因为对于每一层的更新情况不一样,主要是层数奇偶性不同的更新情况不一样,我们可以建立两棵线段树,一个记录的是奇数层的节点的情况,一个是偶。当为奇数线段树的时候,不用管偶数层的节点。偶数线段树同理。
具体可以看一下代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 10;
typedef long long ll;
#define clr(x,y) memset(x,y,sizeof x)
#define INF 0x3f3f3f3f
int a[maxn];
struct Edge{int to,next;}edge[maxn << 2];
int head[maxn],edge_num,num;
int dep[maxn],id[maxn],R[maxn],deps[maxn];
int p[maxn];
void Init()
{
clr(head,-1);edge_num = 0;num = 0;
}
void add_edge(int x,int y)
{
edge[edge_num] = (Edge){y,head[x]};head[x] = edge_num ++;
}
void dfs(int u,int pre,int d)
{
dep[u] = d;id[u] = ++ num;p[num] = u;deps[num] = d;
for(int i = head[u];i != -1;i = edge[i].next)
{
int v = edge[i].to;
if(v == pre)continue;
dfs(v,u,d + 1);
}
R[u] = num;
}
int add[maxn << 2][2];
void build(int l,int r,int rt)
{
add[rt][0] = add[rt][1] = 0;
if(l == r)return;
int mid = (l + r) >> 1;
build(l,mid,rt << 1);build(mid + 1,r,rt << 1|1);
}
void push_down(int rt)
{
if(add[rt][0])
{
add[rt << 1][0] += add[rt][0];
add[rt << 1|1][0] += add[rt][0];
add[rt][0] = 0;
}
if(add[rt][1])
{
add[rt << 1][1] += add[rt][1];
add[rt << 1|1][1] += add[rt][1];
add[rt][1] = 0;
}
}
void update(int L,int R,int x,int l,int r,int rt)
{
if(L <= l && R >= r)
{
add[rt][0] -= x;add[rt][1] += x;return ;
}
push_down(rt);
int mid = (l + r) >> 1;
if(L <= mid)
update(L,R,x,l,mid,rt <<1);
if(R >= mid + 1)
update(L,R,x,mid + 1,r,rt <<1|1);
}
int query(int pos,int l,int r,int rt)
{
if(l == r)
{
return add[rt][deps[l]%2] + a[p[l]];
}
push_down(rt);
int mid = (l + r) >> 1;
if(pos <= mid)
return query(pos,l,mid,rt << 1);
else return query(pos,mid + 1,r,rt << 1|1);
}
int main()
{
int n,m;
while( ~ scanf("%d%d",&n,&m))
{
Init();
for(int i = 1;i <= n;i ++)scanf("%d",&a[i]);
for(int i = 1;i <= n - 1;i ++)
{
int x,y;scanf("%d%d",&x,&y);add_edge(x,y);add_edge(y,x);
}
dfs(1,1,1);
build(1,n,1);
while(m --)
{
int type;scanf("%d",&type);
if(type == 1)
{
int x,y;scanf("%d%d",&x,&y);
if(dep[x] % 2 == 0)y = -y;
update(id[x],R[x],y,1,num,1);
}
else
{
int x;scanf("%d",&x);
printf("%d\n",query(id[x],1,num,1));
}
}
}
return 0;
}