CodeForces 877E DFS序+线段树
题意
就是树上有n个点,然后每个点都有一盏灯,给出初始的状态,1表示亮,0表示不亮,然后有两种操作,第一种是get x,表示你需要输出x的子树和x本身一共有几个灯是亮的。pow x,表示你需要改变x的子树和x本身上的灯的状态。
题解思路
这个题肯定是用DFS序了,为啥?因为树不好操作啊(我也不会啊),使用DFS序可以把树压成一维的一串数,这样就可以使用线段树来进行区间操作了。
话说这个题是我暑假限时训练中做的,看到这个题老开心了,但是让我万万没想到的是,这个题我交了11次才过,太心酸了。
代码实现
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<iostream>
#include<vector>
using namespace std;
const int maxn=2e5+100;
struct node{
int l, r;
int sum;
int lazy;
}t[maxn<<2];
vector<int> g[maxn];
int num[maxn];
int in[maxn], out[maxn], rk[maxn], cnt;
int n, m;
void dfs(int u, int fa) //
{
in[u]=++cnt;//这个是u进去的是时间
rk[cnt]=u; //不要忘了这个,这个是新生成的序列,也就是树被压成一维的那个数列。就是这忘了,害的我好苦。
int len=g[u].size();
for(int i=0; i<len; i++)
{
int v=g[u][i];
if(v==fa) continue;
dfs(v, u);
}
out[u]=cnt;
}
void up(int rt)
{
t[rt].sum=t[rt<<1].sum+t[rt<<1|1].sum;
}
void build(int rt, int l, int r)
{
t[rt].l=l;
t[rt].r=r;
t[rt].sum=0;
t[rt].lazy=0;
if(l==r)
{
t[rt].sum=num[rk[l]]; //这里也不要能错了,很重要。
return ;
}
int mid=(l+r)>>1;
build(rt<<1, l, mid);
build(rt<<1|1, mid+1, r);
up(rt);
}
void down(int rt)
{
if(t[rt].lazy==0) return ;
int ls=rt<<1, rs=rt<<1|1;
t[ls].lazy=!t[ls].lazy;
t[ls].sum=t[ls].r-t[ls].l+1-t[ls].sum;
t[rs].lazy=!t[rs].lazy;
t[rs].sum=t[rs].r-t[rs].l+1-t[rs].sum;
t[rt].lazy=0;
}
void update(int rt, int l, int r)
{
if(l <= t[rt].l && t[rt].r <= r )
{
t[rt].lazy = !t[rt].lazy;
t[rt].sum=t[rt].r-t[rt].l+1-t[rt].sum;
return ;
}
down(rt);
int mid=(t[rt].l+t[rt].r)>>1;
if(l<=mid) update(rt<<1, l, r);
if(r>mid) update(rt<<1|1, l, r);
up(rt);
}
int query(int rt, int l, int r)
{
if(l <= t[rt].l && t[rt].r <= r)
{
return t[rt].sum;
}
int ans=0;
down(rt);
int mid=(t[rt].l+t[rt].r)>>1;
if(l<=mid) ans+=query(rt<<1, l, r);
if(r>mid) ans+=query(rt<<1|1, l, r);
return ans;
}
void init()
{
cnt=0;
for(int i=0; i<=n; i++)
{
g[i].clear();
}
}
int main()
{
scanf("%d", &n);
{
cnt=0;
int x;
string op;
if(n==1)
{
scanf("%d", &x);
scanf("%d" , &m);
while(m--)
{
int tmp;
cin>>op;
scanf("%d", &tmp);
if(op=="get") printf("%d\n", x);
else x=!x;
}
return 0;
}
for(int i=2; i<=n; i++)
{
scanf("%d", &x);
g[x].push_back(i);
g[i].push_back(x);
}
dfs(1, 0);
for(int i=1; i<=n; i++)
{
scanf("%d", &num[i]);
}
build(1, 1, n);
scanf("%d", &m);
for(int i=1; i<=m; i++)
{
cin>>op;
scanf("%d", &x);
if(op=="get")
{
printf("%d\n", query(1, in[x], out[x]));
}
else
{
update(1, in[x], out[x]);
}
}
}
return 0;
}