C. Propagating tree

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

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

可以分奇偶性考虑

奇数 +-+-, -+-+

偶数 +-+, -+-

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

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

查询的时候奇数点直接+au,偶数点相反再+au

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;
}
相关推荐
念何架构之路1 小时前
zap采样器与性能优化内幕
算法·性能优化·哈希算法
晒太羊的猫2 小时前
hot100动态规划刷题
算法·动态规划
weixin_531094692 小时前
heap_4内存管理
java·开发语言·算法
余额瞒着我当琳2 小时前
红黑树的实现--红黑树的概念、规则、效率、结构、旋转、查找、验证
数据结构·c++·算法
程序员南飞2 小时前
win系统安装minio
java·数据结构·算法
青少儿编程课堂2 小时前
AC自动机多模式串匹配精讲:校园关键词巡检实战解析
c++·python·算法·bfs·信息学竞赛
longlongzihan3 小时前
LeetCode 5. 最长回文子串:从暴力匹配到中心扩展的思维演进
c++·算法·leetcode
waoooqwe3 小时前
问卷样本真实吗
大数据·数据库·算法·数据分析
苦瓜打怪兽3 小时前
leetcode 热题100(11. 盛最多水的容器)
python·算法·leetcode
w_zero_one4 小时前
链表(4)
java·数据结构·算法