题意: 两种操作,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; }