Codeforces 383C Propagating tree DFS序+BIT

      一颗以1为根节点的树,两种操作,1 x c 将节点x值+c,节点x的所有孩子的值-c,所有孩子的孩子的节点+c...交替正负直到叶子节点;2 x 查询节点x当前的值。挺简单的一题..当时怎么就没去看...这题只要能想到时间戳的话,就简单了..首先dfs遍历树打上时间戳,这样每个节点保存一个区间l,r表示访问到该节点到遍历完其子树的区间.每个节点按其到根节点的距离的奇偶性分类,ct[i]=0,1表示其奇偶性。用两个树状数组bit[1]表示奇点的操作,bit[2]表示偶点的操作,这样,再执行1操作的时候,根据ct[x]的值,在指定的bit种对区间[Lx,Rx]+c,在另一个bit上对[Lx,Rx]-c。查询的时候,输出初始值+Bit[ct[x]].query(x)就ok。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <vector>
using namespace std;
typedef long long ll;
const int maxn=200000+200;
int mark;
int n,m;
struct BIT
{
    int dt[maxn];
    void init()
    {
        memset(dt,0,sizeof dt);
    }
    int lowbit(int x)
    {
        return (-x)&x;
    }
    void modify(int x,int c)
    {
        if (x==0 || x>n) return;
        for (int x1=x; x1<=n; x1+=lowbit(x1))
        {
            dt[x1]+=c;
        }
    }
    int query(int x)
    {
        int res=0;
        for (int x1=x; x1>0; x1-=lowbit(x1))
        {
            res+=dt[x1];
        }
        return res;
    }
}bit[2];
int a[maxn];
vector<int> g[maxn];
int l[maxn],r[maxn];
int ct[maxn];
void dfs(int u,int fa,int &mark,int cnt)
{
    int v;
    ct[u]=cnt;
    mark++;
    l[u]=mark;
    for (int i=0; i<g[u].size(); i++)
    {
        v=g[u][i];
        if (v!=fa)
        {
            dfs(v,u,mark,cnt^1);
        }
    }
    r[u]=mark;
}
int main()
{
//    freopen("in.txt","r",stdin);
    scanf("%d%d",&n,&m);
    memset(l,0,sizeof l);
    memset(r,0,sizeof r);

    for (int i=1; i<=n; i++)
    scanf("%d",&a[i]);
    int x,y,z;
    for (int i=1; i<=n-1; i++)
    {
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    mark=0;
    dfs(1,0,mark,0);
    r[1]=mark;
//    for (int i=1; i<=n; i++)
//    cout<<l[i]<<" "<<r[i]<<endl;
    bit[0].init();
    bit[1].init();
    for (int i=1; i<=m; i++)
    {
        scanf("%d",&z);
        if (z==1)
        {
            scanf("%d%d",&x,&y);
            bit[ct[x]].modify(l[x],y);
            bit[ct[x]].modify(r[x]+1,-y);

            bit[ct[x]^1].modify(l[x],-y);
            bit[ct[x]^1].modify(r[x]+1,y);
        }
        else
        {
            if (z==2)
            {
                scanf("%d",&x);
                int out=a[x]+bit[ct[x]].query(l[x]);
                printf("%d\n",out);
            }
        }
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值