Codeforces Round #442 (Div. 2) E. Danil and a Part-time Job 【线段树】

最新推荐文章于 2022-07-13 02:01:45 发布

学不会dp QAQ

最新推荐文章于 2022-07-13 02:01:45 发布

阅读量116

点赞数

分类专栏：数据结构文章标签：数据结构

本文链接：https://blog.csdn.net/qq_50554695/article/details/120735344

版权

数据结构专栏收录该内容

25 篇文章 1 订阅

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）构建的线段树数据结构，以解决关于以1为根节点的树中灯状态切换及亮灯数量查询的问题。通过DFS获取节点区间，并利用线段树进行高效操作，适用于实时更新和查询子树灯状态的应用场景。

摘要由CSDN通过智能技术生成

题目链接

题意

给出一棵以1为根结点树，每个结点有一个灯，进行两种操作：

改变结点 $x$ 及其子树的灯的状态（开变关，关变开）
查询结点 $x$ 及其子树中亮着的灯数

题解

以DFS序建立线段树，每个子树都是一段连续区间，然后就是线段树基本操作了

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<list>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#include<random>
using namespace std;
#include<ext/pb_ds/priority_queue.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
using namespace __gnu_pbds;
#include<ext/rope>
using namespace __gnu_cxx;

#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))

const int mod = 1e9+7;
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}
const int INF = 0x3f3f3f3f;
const int N = 1e6+10;
struct node{
	int l,r,mid;
	int w,tag;
}seg[N];
vector<int>g[N];
int a[N],dfn[N],in[N],out[N],tim;
void dfs(int u,int fa){
	in[u]=++tim,dfn[tim]=u;
	for(auto v:g[u]){
		if(v==fa)continue;
		dfs(v,u);
	}
	out[u]=tim;
}
void pu(int rt){seg[rt].w=seg[rt<<1].w+seg[rt<<1|1].w;}
void pd(int rt){
	seg[rt<<1].tag^=1;
	seg[rt<<1].w=(seg[rt<<1].r-seg[rt<<1].l+1)-seg[rt<<1].w;
	seg[rt<<1|1].tag^=1;
	seg[rt<<1|1].w=(seg[rt<<1|1].r-seg[rt<<1|1].l+1)-seg[rt<<1|1].w;
	seg[rt].tag=0;
}
void build(int rt,int l,int r){
	int mid=(l+r)>>1;
	seg[rt]={l,r,mid,0,0};
	if(l==r){seg[rt].w=a[dfn[l]];return;}
	build(rt<<1,l,mid),build(rt<<1|1,mid+1,r);
	pu(rt);
}
void update(int rt,int l,int r){
	if(seg[rt].l>r||seg[rt].r<l)return;
	if(l<=seg[rt].l&&seg[rt].r<=r){
		seg[rt].w=(seg[rt].r-seg[rt].l+1)-seg[rt].w;
        seg[rt].tag^=1;
        return ;
	}
	if(seg[rt].tag)pd(rt); 
	if(seg[rt].mid>=l)update(rt<<1,l,r);
	if(seg[rt].mid<r)update(rt<<1|1,l,r);
	pu(rt);
}
int query(int rt,int l,int r){
	if(seg[rt].l>r||seg[rt].r<l)return 0;
	if(l<=seg[rt].l&&seg[rt].r<=r)return seg[rt].w;
	int ans=0;
	if(seg[rt].tag)pd(rt); 
	if(seg[rt].mid>=l)ans+=query(rt<<1,l,r);
	if(seg[rt].mid<r)ans+=query(rt<<1|1,l,r);
	pu(rt);
	return ans;
}
#define endl '\n'
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	int n; cin>>n;
	for(int i=2;i<=n;i++){
		int v; cin>>v;
		g[i].push_back(v);
		g[v].push_back(i);
	}
	for(int i=1;i<=n;i++)cin>>a[i];
	dfs(1,1);
	build(1,1,n);
	int q; cin>>q;
	while(q--){
		string op;int x; cin>>op>>x;
		if(op=="get")cout<<query(1,in[x],out[x])<<endl;
		if(op=="pow")update(1,in[x],out[x]);
	}
}