场景二:参考代码C++

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <cmath>
#define ll long long
using namespace std;
const int N=1e5+10,mod=1e9+7,INF=0x3f3f3f3f;
vector<int> G[N]; 
int l[N],r[N],pos[N],dfs_clock;
ll val[N];
struct node {
	int Left,Right;
	ll sum;
}NODE[N<<2];
void PushUp(int idx) {
	NODE[idx].sum=NODE[idx<<1].sum+NODE[idx<<1|1].sum;
}
void build(int l, int r, int idx) {
	NODE[idx]={l,r,0};
	if (l==r) {
		NODE[idx].sum=val[pos[l]];
		return;
	}
	int m=(l+r)>>1;
	build(l,m,idx<<1),build(m+1,r,idx<<1|1);
	PushUp(idx);
}
void Updata(int pos, int idx, int val) {
	if (NODE[idx].Left==NODE[idx].Right) {
		NODE[idx].sum=val;
		return;
	}
	int m=(NODE[idx].Left+NODE[idx].Right)>>1;
	if (pos<=m) Updata(pos,idx<<1,val);
	else Updata(pos,idx<<1|1,val);
	PushUp(idx);
}
ll query(int l, int r, int idx) {
	if (NODE[idx].Left==l&&NODE[idx].Right==r) return NODE[idx].sum;
	int m=(NODE[idx].Left+NODE[idx].Right)>>1;
	if (r<=m) return query(l,r,idx<<1);
	else if (l>m) return query(l,r,idx<<1|1);
	else return query(l,m,idx<<1)+query(m+1,r,idx<<1|1);
}
void dfs(int u, int fa) {
	l[u]=++dfs_clock;
	pos[dfs_clock]=u; 
	for (int i=0; i<G[u].size(); i++) {
		int v=G[u][i];
		if (v==fa) continue;
		dfs(v,u);
	}
	r[u]=dfs_clock;
}
int main() {
	int n,q; scanf("%d%d",&n,&q);
	for (int i=1; i<n; i++) {
		int u,v; scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	for (int i=1; i<=n; i++) scanf("%lld",&val[i]);
	dfs(1,0);
	build(1,n,1);
	while (q--) {
		int op; scanf("%d",&op);
		if (op==1) {
			int x,y; scanf("%d%d",&x,&y);
			Updata(l[x],1,y);
			val[x]=y;
		} else {
			int x; scanf("%d",&x);
			printf("%lld\n",query(l[x],r[x],1));
		}
	}
    return 0;
} 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值