牛客——求和(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;
}
目录
相关文章
|
9月前
|
移动开发 算法 机器人
[蓝桥杯] 二分与前缀和习题练习
又来更新了。二分与前缀和是蓝桥杯比较常考的内容,所以我们要多加练习这方面的习题。二分与前缀和的难度相对来说不算大,我们应该掌握其关键要点。
77 0
|
10月前
蓝桥杯:桶排序 与 例题:算式问题
蓝桥杯:桶排序 与 例题:算式问题
53 0
|
11月前
|
存储 人工智能 测试技术
【AcWing每日一题】4644. 求和
【AcWing每日一题】4644. 求和
59 0
|
9月前
|
机器学习/深度学习 存储 C++
[蓝桥杯] 树状数组与线段树问题(C/C++)
[蓝桥杯] 树状数组与线段树问题(C/C++)
60 0
|
10月前
|
算法
数星星(树状数组模板题)
数星星(树状数组模板题)
30 0
每日一题——有序数组的平方
每日一题——有序数组的平方
|
存储
蓝桥杯 ADV_303 数组求和
蓝桥杯 ADV_303 数组求和
62 0
每日一题:Leetcode977 有序数组的平方
每日一题:Leetcode977 有序数组的平方
|
人工智能 索引
树状数组总结
能够解决数据压缩里的累积频率的计算问题,现多用于高效计算数列的前缀和、区间和。
108 0
树状数组总结