传送门:牛客
题目描述:
已知有 n 个节点,有 n−1 条边,形成一个树的结构。给定一个根节点
k,每个节点都有一个权值,节点i的权值为 v i给 m 个操作,操作有两种类型:
1 a x :表示将节点 a 的权值加上 x
2 a :表示求 a 节点的子树上所有节点的和(包括 a 节点本身)
输入:
5 6 1
1 2 3 4 5
1 3
1 2
2 4
2 5
1 2 10
1 3 10
1 4 5
1 5 1
2 3
2 2
输出:
13
27
树上的单点修改+区间求和,使用树链剖分+线段树进行解决即可
使用树链剖分将树形结构分解为数组线性结构,然后使用线段树进行维护即可
对于树链剖分+线段树,网上有大量题解对此进行讲解,次数就不再赘述了
下面是具体的代码部分:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
ll x=0,w=1;char ch=getchar();
for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x*w;
}
#define int long long
#define maxn 1000100
const double eps=1e-8;
#define int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
int fa[maxn],Size[maxn],max_son[maxn],dep[maxn];
vector<int>edge[maxn];
void dfs1(int u,int per_u) {
Size[u]=1;
for(int i=0;i<edge[u].size();i++) {
int v=edge[u][i];
if(v==per_u) continue;
dep[v]=dep[u]+1;
dfs1(v,u);
fa[v]=u;Size[u]+=Size[v];
if(Size[v]>Size[max_son[u]]) {
max_son[u]=v;
}
}
}
int top[maxn],id[maxn],rev[maxn],tot=0;
void dfs2(int u,int t) {
top[u]=t;id[u]=++tot;rev[tot]=u;
if(!max_son[u]) return ;
dfs2(max_son[u],t);
for(int i=0;i<edge[u].size();i++) {
int v=edge[u][i];
if(v==fa[u]||v==max_son[u]) continue;
dfs2(v,v);
}
}
struct Segment_tree{
int l,r,sum;
}tree[maxn*4];int w[maxn];
void pushup(int rt) {
tree[rt].sum=tree[ls].sum+tree[rs].sum;
}
void build(int l,int r,int rt) {
tree[rt].l=l;tree[rt].r=r;
if(l==r) {
tree[rt].sum=w[rev[l]];
return ;
}
int mid=(l+r)>>1;
build(lson);build(rson);
pushup(rt);
}
void update(int pos,int rt,int val){
if(tree[rt].l==pos&&tree[rt].r==pos) {
tree[rt].sum+=val;
return ;
}
int mid=(tree[rt].l+tree[rt].r)>>1;
if(pos<=mid) update(pos,ls,val);
else update(pos,rs,val);
pushup(rt);
}
int ans_sum=0;
void query(int l,int r,int rt) {
if(tree[rt].l==l&&tree[rt].r==r) {
ans_sum+=tree[rt].sum;
return ;
}
int mid=(tree[rt].l+tree[rt].r)>>1;
if(r<=mid) query(l,r,ls);
else if(l>mid) query(l,r,rs);
else query(l,mid,ls),query(mid+1,r,rs);
}
void ask(int u) {
query(id[u],id[u]+Size[u]-1,1);
}
int n,m,k;
signed main() {
n=read();m=read();k=read();
for(int i=1;i<=n;i++) w[i]=read();
for(int i=1;i<=n-1;i++) {
int x=read(),y=read();
edge[x].push_back(y);
edge[y].push_back(x);
}
dfs1(k,0);dfs2(k,k);
build(1,n,1);
for(int i=1;i<=m;i++) {
int opt=read();
if(opt==1) {
int u=read(),val=read();
update(id[u],1,val);
}
else {
int u=read();
ans_sum=0;
ask(u);
printf("%lld\n",ans_sum);
}
}
return 0;
}