[codeforces] 383C Propagating tree(dfs序+线段树)

Propagating tree

https://ac.nowcoder.com/acm/problem/110318

题意:
给你一棵n个结点的树, 以1为根。每个结点有点权。有m次操作:
1.x结点权值 +val,x的儿子权值 −val,x的孙子们 +val,以此类推。
2.询问x的点权;
题解:
我们首先跑一边dfs序,顺便求除每个结点的深度。
我们把他分成两颗线段树,深度为奇数的在第一颗上,深度为偶数的在第二课上。
我们每次对两颗线段树同时操作。
第一颗线段树只进行+操作,第二棵线段树只-操作。
当我们询问某个点时,我们只需要判断一下这个点的奇偶,决定查询哪颗树即可。

/*Keep on going Never give up*/
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
//#define int long long

#define endl '\n'
#define Accepted 0
#define AK main()
#define I_can signed
using namespace std;
const int maxn =4e5+10;
const int MaxN = 0x3f3f3f3f;
const int MinN = 0xc0c0c00c;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll mod=1e9+7;

vector<int> edge[maxn];
int a[maxn];
int in[maxn],out[maxn],cnt;
int tree[2][maxn<<2];
int deep[maxn];
void dfs(int u,int fa){
    in[u]=++cnt;

    deep[u]=deep[fa]+1;
    for(auto i:edge[u]){
        if(i==fa) continue;
        dfs(i,u);
    }
    out[u]=cnt;
}

void push(int node,int x){
    if(tree[x][node]){
        tree[x][node*2]+=tree[x][node];
        tree[x][node*2+1]+=tree[x][node];
        tree[x][node]=0;
    }
    return;
}

void update(int node,int l,int r,int start,int ends,int x,int val){
    if(start<=l&&ends>=r){
        tree[x][node]+=val;
        //cout<<tree[x][node]<<endl;
        return;
    }
    push(node,x);
    int mid=(l+r)/2;
    if(start<=mid) update(node*2,l,mid,start,ends,x,val);
    if(mid<ends) update(node*2+1,mid+1,r,start,ends,x,val);
}

int query(int node,int l,int r,int pos,int x){
    if(l==r){
        //cout<<tree[x][node]<<endl;
        return tree[x][node];

    }
    push(node,x);
    int mid=(l+r)/2;
    if(pos<=mid) return query(node*2,l,mid,pos,x);
    else return query(node*2+1,mid+1,r,pos,x);
}

signed main(){
//
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=0;i<n-1;i++){
        int x,y;
        cin>>x>>y;
        edge[x].push_back(y);
        edge[y].push_back(x);
    }
    dfs(1,0);
    //for(int i=1;i<=n;i++) cout<<in[i]<<" ";

    for(int i=0;i<m;i++){
        int opt,x,y;
        cin>>opt;
        if(opt==1){          //246   296
            cin>>x>>y;
            //cout<<in[x]<<" "<<out[x]<<endl;
            update(1,1,n,in[x],out[x],deep[x]&1,y);
            update(1,1,n,in[x],out[x],!(deep[x]&1),-y);
        }
        else{
            cin>>x;

            cout<<query(1,1,n,in[x],deep[x]&1)+a[x]<<endl;
        }
    }
    return 0;
}
题解 文章被收录于专栏

主要写一些题目的题解

全部评论

相关推荐

02-14 07:38
已编辑
门头沟学院 Java
2.4&nbsp;一面2.6&nbsp;二面2.9&nbsp;三面(hr面)2.13&nbsp;oc1.15号收到面试电话那会就开始准备,因为一开始没底所以选择推迟一段时间面试,之后开始准备八股,准备实习可能会问的东西,这期间hot100过了有六七遍,真的是做吐了快,八股也是背了忘,忘了背,面经也看了很多,虽然最后用上的只有几道题,可是谁知道会问什么呢自从大二上开始学java以来,一路走来真的太痛了,一开始做外卖,点评,学微服务,大二下五六月时,开始投简历,哎,投了一千份了无音讯,开始怀疑自己(虽然能力确实很一般),后来去到一家小小厂,但是并不能学到什么东西,而且很多东西都很不规范,没待多久便离开,大二暑假基本上摆烂很怀疑自己,大三上因为某些原因开始继续学,期间也受到一俩个中小厂的offer,不过学校不知道为啥又不允许中小厂实习只允许大厂加上待遇不太好所以也没去,感觉自己后端能力很一般,于是便打算转战测开,学习了一些比较简单的测试理论(没有很深入的学),然后十二月又开始继续投,java和测开都投,不过好像并没有几个面试,有点打击不过并没有放弃心里还是想争一口气,一月初因为学校事比较多加上考试便有几天没有继续投,10号放假后便继续,想着放假应该很多人辞职可能机会大一点,直到接到字节的面试,心里挺激动的,总算有大厂面试了,虽然很开心,但同时压力也很大,心里真的很想很想很想进,一面前几天晚上都睡不好觉,基本上都是二三点睡六七点醒了,好在幸运终于眷顾我一次了(可能是之前太痛了),一面三十几分钟结束,问的都不太难,而且面试官人挺好但是有些问题问的很刁钻问到了测试的一些思想并不是理论,我不太了解这方面,但是也会给我讲一讲他的理解,但是面完很伤心觉得自己要挂了。但是幸运的是一面过了(感谢面试官),两天后二面,问的同样不算难,手撕也比较简单,但也有一两个没答出来,面试官人很好并没有追问,因为是周五进行的二面,没有立即出结果,等到周一才通知到过了,很煎熬的两天,根本睡不好,好在下周一终于通知二面过了(感谢面试官),然后约第二天三面,听别的字节同学说hr面基本上是谈薪资了,但是我的并不是,hr还问了业务相关的问题,不过问的比较浅,hr还问我好像比较紧张,而且hr明确说了还要比较一下,我说我有几家的面试都拒了就在等字节的面试(当然紧张,紧张到爆了要),三面完后就开始等结果,这几天干啥都没什么劲,等的好煎熬,终于13号下午接到了电话通知oc了,正式邮件也同时发了,接到以后真的不敢信,很激动但更重要的是可以松一口气了,可以安心的休息一下了终于可以带着个好消息过年了,找实习也可以稍微告一段落了,虽然本人很菜,但是感谢字节收留,成为忠诚的节孝子了因为问的比较简单,面经就挑几个记得的写一下一面:1.实习项目的难点说一下2.针对抖音评论设计一下测试用例3.手撕:合并两个有序数组二面:1.为什么转测开2.线程进程区别,什么场景适合用哪个3.发送一个朋友圈,从发出到别人看到,从数据流转的角度说一下会经历哪些过程4.针对抖音刷到广告视频设计测试用例5.手撕:无重复字符的最长字串
牛客85811352...:测开问这么简单?
查看8道真题和解析
点赞 评论 收藏
分享
评论
2
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务