Codeforces 383C Propagating tree(树状数组）

最新推荐文章于 2021-06-09 22:36:11 发布

Rain722

最新推荐文章于 2021-06-09 22:36:11 发布

阅读量401

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构-线段树

本文链接：https://blog.youkuaiyun.com/Rain722/article/details/78123474

数据结构-线段树专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种使用树状数组结合深度优先搜索（DFS）算法来解决涉及树形结构的操作问题的方法。具体而言，对于一棵树，文章讨论了如何高效地执行两种操作：一是对指定节点及其子孙节点进行特定值的加减操作；二是查询指定节点的当前值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：

有一棵树，对这个树有两种操作：1：表示为（1 x val），在编号为x的节点上加上val，然后给x节点的每个儿子加上- val，再给每个儿子的儿子加上-（- val），一直加到没有儿子为止。2：表示为（2 x）查询x节点上的值。

思路：

首先通过dfs将树抽象成树状数组，然后以和根节点的层数关系作为d，开两个树状数组分别计算添加值和减少值。

#include<bits/stdc++.h>
using namespace std;
#define N 200005
int bit[2][N], cnt, n;
struct Node
{
    int l, r, val, d;
}node[N];
vector<int> g[N];
void dfs(int u, int fa, int d)
{
    node[u].l=++cnt;node[u].d=d;
    for(int i=0; i<g[u].size(); i++)
    {
        int v=g[u][i];
        if(v==fa) continue;
        dfs(v, u, 1-d);
    }
    node[u].r=cnt;
}
void modify(int x, int val, int *sum)
{
    while(x<=n)
    {
        sum[x]+=val;
        x+=(x&(-x));
    }
}
int getsum(int x, int *sum)
{
    int ret=0;
    while(x>0)
    {
        ret+=sum[x];
        x-=(x&(-x));
    }
    return ret;
}
int main()
{
    int m, u, v, op;
    while(scanf("%d%d", &n, &m)!=EOF)
    {
        cnt=0;
        memset(bit, 0, sizeof(bit));
        for(int i=1; i<=n; i++)
        {
            scanf("%d", &node[i].val);
            g[i].clear();
        }
        for(int i=1; i<n; i++)
        {
            scanf("%d%d", &u, &v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        dfs(1, -1, 0);
        for(int i=1; i<=m; i++)
        {
            scanf("%d", &op);
            if(op==1)
            {
                scanf("%d%d", &u, &v);
                modify(node[u].l, v, bit[node[u].d]);
                modify(node[u].r+1, -v, bit[node[u].d]);
                modify(node[u].l, -v, bit[1-node[u].d]);
                modify(node[u].r+1, v, bit[1-node[u].d]);
            }
            else
            {
                scanf("%d", &u);
                printf("%d\n", node[u].val+getsum(node[u].l, bit[node[u].d]));
            }
        }
    }
    return 0;
}