C. Propagating tree

https://codeforces.com/contest/383/problem/C

因为是单点查询,操作一是+-+-

可以分奇偶性考虑

奇数 +-+-, -+-+

偶数 +-+, -+-

我们可以发现奇数和偶数的标记是相反的

因此我们维护奇数点的标记,如果点是奇数就+val,否者就-val

查询的时候奇数点直接+a[u],偶数点相反再+a[u]

code

cpp 复制代码
// Problem: C. Propagating tree
// Contest: Codeforces - Codeforces Round 225 (Div. 1)
// URL: https://codeforces.com/problemset/problem/383/C
// Memory Limit: 256 MB
// Time Limit: 2000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<iostream>
#include<algorithm>
#include<vector>
#define INF (1ll<<60)
using namespace std;
typedef long long ll;
const int N=2e5+9;
int a[N],fa[N],sz[N],wc[N],dep[N],dfn[N],rdfn[N],top[N];
int n,m,r,p,vistime;
vector<int> e[N<<1];
void add(int u,int v){
    e[u].push_back(v);
    e[v].push_back(u);
}
void dfs1(int u,int f){
    fa[u]=f;
    sz[u]=1;
    dep[u]=dep[f]+1;
    for(auto & i : e[u]){
        if(i!=f){
            dfs1(i,u);
            sz[u]+=sz[i];
            if(sz[i]>sz[wc[u]]){
                wc[u]=i;
            }
        }
    }
}
void dfs2(int u,int Top){
    dfn[u]=++vistime;
    rdfn[vistime]=u;
    top[u]=Top;
    if(wc[u]){
        dfs2(wc[u],Top);
        for(auto & i : e[u]){
            if(i!=wc[u] && i!=fa[u]){
                dfs2(i,i);
            }   
        }
    }
}
struct node{
    ll val,tag;
}seg[N<<2];
ll tl(ll x){
    return x<<1;
}
ll tr(ll x){
    return x<<1|1;
}
void pushup(int id){
    seg[id].val=(seg[tl(id)].val+seg[tr(id)].val);
}
bool inrange(int L,int R,int l,int r){
    return L>=l && R<=r;
}
bool outofrange(int L,int R,int l,int r){
    return  L>r || l>R;
}
void maketag(int id,int l,int r,ll v){
    (seg[id].val+=(r-l+1)*v);
    (seg[id].tag+=v);
}
void pushdown(int id,int l,int r){
    int mid=(l+r)>>1;
    maketag(tl(id),l,mid,seg[id].tag);  
    maketag(tr(id),mid+1,r,seg[id].tag);
    seg[id].tag=0;  
}
ll query(int id,int L,int R,int l,int r){
    if(inrange(L,R,l,r)){
        return seg[id].val;
    }else if(!outofrange(L,R,l,r)){
        int mid=(L+R)>>1;
        pushdown(id,L,R);
        return (query(tl(id),L,mid,l,r)+query(tr(id),mid+1,R,l,r));
    }else{
        return 0;
    }
}
void modify(int id,int L,int R,int l,int r,ll v){
    if(inrange(L,R,l,r)){
        maketag(id,L,R,v);
    }else if(!outofrange(L,R,l,r)){
        int mid=(L+R)>>1;
        pushdown(id,L,R);
        modify(tl(id),L,mid,l,r,v);
        modify(tr(id),mid+1,R,l,r,v);
        pushup(id);
    }
}
ll ask(int id,int l,int r,int pos){
	if(l==r){
		return seg[id].val;
	}
	int mid=(l+r)>>1;
	pushdown(id,l,r);
	if(mid>=pos){
		return ask(tl(id),l,mid,pos);
	}else{
		return ask(tr(id),mid+1,r,pos);
	}
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	for(int i=1;i<=n-1;i++){
		int u,v;
		cin>>u>>v;
		add(u,v);
	}
	//root=1
	dfs1(1,0);
	dfs2(1,0);
	for(int i=1;i<=m;i++){
		int op;
		cin>>op;
		if(op==1){
			int x,val;
			cin>>x>>val;
			val*=(dep[x]&1?1:-1);
			modify(1,1,n,dfn[x],dfn[x]+sz[x]-1,val);
		}else{
			int x;
			cin>>x;
			ll ans=ask(1,1,n,dfn[x]);
			ans*=(dep[x]&1?1:-1);
			ans+=a[x];
			cout<<ans<<'\n';
		}
	}
    return 0;
}
相关推荐
鱼跃鹰飞44 分钟前
Leetcode面试经典150题-141.环形链表
算法·leetcode·链表·面试
吕小明么1 小时前
北大&阿里:新出炉的LLM偏好对齐方法综述
人工智能·算法·语言模型·aigc·agi
py.鸽鸽1 小时前
C语言——数组,指针,指针数组,数组指针
c语言·算法
理想青年宁兴星1 小时前
【数据结构】字符串与JSON字符串、JSON字符串及相应数据结构(如对象与数组)之间的相互转换
java·数据结构·json
立志成为coding大牛的菜鸟.6 小时前
力扣139-单词拆分(Java详细题解)
java·算法·leetcode
星夜孤帆7 小时前
LeetCode之数组/字符串
java·算法·leetcode
present12277 小时前
利用matlab bar函数绘制较为复杂的柱状图,并在图中进行适当标注
算法·matlab·数据分析·学习方法
就这样很好88010 小时前
排序算法总结
java·算法·排序算法
weixin_4866811410 小时前
C++系列-STL中find相关的算法
java·c++·算法
我是真爱学JAVA10 小时前
第四章 类和对象 课后训练(1)
java·开发语言·算法