Propagating tree(奇偶性+dfs序+差分线段树)

Propagating tree

https://ac.nowcoder.com/acm/problem/110318?&headNav=acm

图片说明
图片说明
题目大意:
给你一颗树,你可以在树上进行两种操作
1 x val,把val加到x节点上面去,然后-val加到x的第一层儿子节点,然后-(-val)加到第二层儿子节点,。。。等
2 x,查询x节点的值。
思路:
这题看了大佬的思路才做出来的。

https://blog.nowcoder.net/n/d8f8de427aae4b65a43bf5383a75938b

这个题的关键就是先把每个节点是奇数层还是偶数层求出来,然后求一个dfs序,把每个节点为子树的儿子节点的范围求出来。
对于1操作,该节点能够影响的范围有[l[x],r[x]],那么对这个区间做一个加法,我们可以用线段树或者树状数组差分来实现s[l[x]]+=v,s[r[x]+1]-=v;
对于2操作,该节点x的值=w[i]+(奇)[1,l[x]]-(偶)[1,l[x]]。
所以我们可以通过维护两颗线段树(维护奇数层和偶数层)来实现。
代码:

#include<iostream>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;

const int maxn = 4e5 + 10;
vector<int>G[maxn];
int in[maxn],out[maxn],tot;
bool cal[maxn];
int w[maxn];
void dfs(int u,int fu,int dep){
    in[u] = ++tot;cal[u] = dep % 2;
    for(int i = 0; i < G[u].size(); i++){
        int v = G[u][i];
        if(v != fu){
            dfs(v,u,dep + 1);
        }
    }
    out[u] = tot;
}
#define lson (rt * 2)
#define rson (rt * 2) + 1
#define mid (l + r) / 2
struct seg{
    int s[maxn << 2];
    void update(int l,int r,int pos,int rt,int v){
        if(l == r){
            s[rt] += v;
            return ;
        }
        if(pos <= mid)update(l,mid,pos,lson,v);
        else update(mid + 1,r,pos,rson,v);
        s[rt] = s[lson] + s[rson];
    }
    int query(int l,int r,int ql,int qr,int rt){
        if(l >= ql && r <= qr){
            return s[rt];
        }
        int ans = 0;
        if(ql <= mid)ans += query(l,mid,ql,qr,lson);
        if(qr > mid)ans += query(mid + 1,r,ql,qr,rson);
        return ans;
    }
}tree[2];
void solved(){
    int n,m;scanf("%d%d",&n,&m);
    for(int i = 1; i <= n; i++)scanf("%d",&w[i]);
    for(int i = 1; i < n; i++){
        int u,v;scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1,0,1);
    ++tot;
    while(m--){
        int ins;scanf("%d",&ins);
        if(ins == 1){
            int x,v;scanf("%d%d",&x,&v);
            tree[cal[x]].update(1,tot,in[x],1,v);
            tree[cal[x]].update(1,tot,out[x] + 1,1,-v);
        }else{
            int x;scanf("%d",&x);
            printf("%d\n",w[x] + tree[cal[x]].query(1,tot,1,in[x],1) - tree[cal[x] ^ 1].query(1,tot,1,in[x],1));
        }
    }
}
int main(){
    solved();
    return 0;
}
牛客每日一题 文章被收录于专栏

牛客每日一题

全部评论

相关推荐

永不遗忘:才这么点算什么拉黑,我初筛连着挂几十次了,最后还是能进面
点赞 评论 收藏
分享
我是985研究生,最近学校在组织开题,大家都在非常紧张地准备,但我一直进入不了状态,很想做但是心又很浮躁。但我的室友们感觉都非常认真,每天醒来就开始看论文,睡着前最后一件事还是在看论文,我非常焦虑。我感觉自己甚至有点把大家当做假想敌了。这种比较心态还存在于生活的各种方面:看到有钱的同学会非常羡慕,看到朋友圈里面环游世界的留学生同学也会羡慕,看到那些工作后有自己的钱而过上较为阔绰的生活的时候还是羡慕,就仿佛只有自己一个人在阴暗爬行。而且这些比较是每时每刻的,为了不比较,我已经关闭了朋友圈,但是每次偶尔刷一下还是会难受很久。我知道比较是偷走幸福的小偷,但我好像控制不了,感觉自己是一个偷窥别人生活的...
若怜君欢:担心开题搞砸了,幻想拥有别人的生活,本质上是因为自卑,楼主小时候大概率是留守儿童或者父母关系很紧张,导致楼主没有安全感、焦虑、内耗。 这样的情况最好的办法就是建立自信和降低期待,建立自信不是一蹴而就,而是循序渐进,比如告诉自己允许自己第一次没把事情做好,失败了能搞清楚其中缘由而不是全盘否定自己,失败不是终点,放弃才是;降低期待只要记住一句话即可,能伴随你一生的,只有经验和学识,所以你对事情的态度应该更多地去思考它是否能带来学识和经验的增长,而不是仅仅用短期的利益作为唯一期待。 人生不是一成不变的,它是可以迭代更新的,去归纳总结自身的不足并结合实际去改进,去尝试一些新的思路和方法,不要固执钻牛角尖,也不要反复横跳,为自己设立一个高度聚集的精神内核,内核之上可以去尝试一切有利于自己更好的方式 以上就是我个人对生活的理解,共勉
点赞 评论 收藏
分享
评论
5
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务