牛客——求和(dfs序+树状数组)

简介: 牛客——求和(dfs序+树状数组)

原题链接

题意:

给定一棵树,有点权;两种操作,一是修改某个单点的权值,二是查询某个点为根节点的子树的点权之和。

思路:

dfs序后,子树都处于一个区间里,所以问题就转化成了单点修改、区间求和。

树状数组维护即可。

代码:

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
const int maxn=1e6+7;
int n,m,k;
int h[maxn],idx;
struct node{
    int e,ne;
}edge[maxn*2];
ll w[maxn],tr[maxn*4];
void add(int u,int v){///前向星存图
    edge[idx]={v,h[u]};h[u]=idx++;
}
ll in[maxn],out[maxn],timetmp;
void dfs(int u,int fa){//dfs记录dfs序列
    in[u]=++timetmp;
    for(int i=h[u];~i;i=edge[i].ne){
        int j=edge[i].e;
        if(j==fa) continue;
        dfs(j,u);
    }
    out[u]=timetmp;
}
///树状数组维护区间
ll lowbit(int x){
    return x&(-x);
}
void update(int pos,int val){
    while(pos<maxn){
        tr[pos]+=val;
        pos=pos+lowbit(pos);
    }
}
ll qask(int pos){
    ll res=0;
    while(pos){
        res+=tr[pos];
        pos=pos-lowbit(pos);
    }
    return res;
}
int main(){
    cin>>n>>m>>k;
    memset(h,-1,sizeof h);
    for(int i=1;i<=n;i++) cin>>w[i];
    for(int i=1;i<n;i++){
        ll u,v;
        cin>>u>>v;
        add(u,v);add(v,u);
    }
    dfs(k,-1);
    for(int i=1;i<=n;i++)
        update(in[i],w[i]);
    while(m--){
        int op;cin>>op;
        if(op==1){
            ll pos,x;
            cin>>pos>>x;
            update(in[pos],x);
        }
        else{
            int pos;cin>>pos;
            printf("%lld\n",qask(out[pos])-qask(in[pos]-1));
        }
    }
    return 0;
}
目录
相关文章
|
6月前
【洛谷 P1873】[COCI 2011_2012 #5] EKO _ 砍树 题解(向量+二分查找)
【COCI 2011/2012 #5】EKO 砍树问题摘要: - 伐木工Mirko需砍$M$米木材,只能砍一排树,使用二分搜索策略确定锯片最大高度$H$。 - 锯片设为$H$米时,会砍掉所有高于$H$的树顶,求得所需木材至少$M$米的最高$H$。 - 输入:树的数量$N$和所需木材总长$M$,每棵树的高度。 - 输出:锯片的最大高度。
44 0
|
存储 人工智能 测试技术
【AcWing每日一题】4644. 求和
【AcWing每日一题】4644. 求和
75 0
LeetCode题:70爬楼梯,126斐波那契数
LeetCode题:70爬楼梯,126斐波那契数
55 0
|
机器学习/深度学习 存储 C++
[蓝桥杯] 树状数组与线段树问题(C/C++)
[蓝桥杯] 树状数组与线段树问题(C/C++)
80 0
|
算法
数星星(树状数组模板题)
数星星(树状数组模板题)
60 0
|
机器学习/深度学习 算法
蓝桥杯:递推算法 递归算法 例题:斐波纳契数列
蓝桥杯:递推算法 递归算法 例题:斐波纳契数列
68 0
每日一题——有序数组的平方
每日一题——有序数组的平方
每日一题:Leetcode977 有序数组的平方
每日一题:Leetcode977 有序数组的平方