[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;
}
题解 文章被收录于专栏

主要写一些题目的题解

全部评论

相关推荐

06-27 12:54
已编辑
门头沟学院 Java
累了,讲讲我的大学经历吧,目前在家待业。我是一个二本院校软件工程专业。最开始选专业是觉得计算机感兴趣,所以选择了他。本人学习计算机是从大二暑假结束开始的,也就是大三开始。当时每天学习,我个人认为Java以及是我生活的一部分了,就这样持续学习了一年半,来到了大四上学期末,大概是在12月中旬,我终于找的到了一家上海中厂的实习,但我发现实习生的工作很枯燥,公司分配的活也不多,大多时间也是自己在自学。就这样我秋招末才找到实习。时间来到了3月中旬,公司说我可以转正,但是转正工资只有7000,不过很稳定,不加班,双休,因为要回学校参加答辩了,同时当时也是心高气傲,认为可以找到更好的,所以放弃了转正机会,回学校准备论文。准备论文期间就也没有投递简历。然后时间来到了5月中旬,这时春招基本也结束了,然后我开始投递简历,期间只是约到了几家下场面试。工资也只有6-7k,到现在我不知道该怎么办了。已经没有当初学习的心劲了,好累呀,但是又不知道该干什么去。在家就是打游戏,boss简历投一投。每天日重一次。26秋招都说是针对26届的人,25怎么办。我好绝望。要不要参加考公、考研、央国企这些的。有没有大佬可以帮帮我。为什么感觉别人找工作都是顺其自然的事情,我感觉自己每一步都在艰难追赶。八股文背了又忘背了又忘,我每次都花很长时间去理解他,可是现在感觉八股、项目都忘完了。真的已经没有力气再去学习了。图片是我的简历,有没有大哥可以指正一下,或者说我应该走哪条路,有点不想在找工作了。
码客明:太累了就休息一下兄弟,人生不会完蛋的
如果实习可以转正,你会不...
点赞 评论 收藏
分享
废物一个0offer:认真的吗二本本科找人工智能岗位
点赞 评论 收藏
分享
评论
2
收藏
分享

创作者周榜

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