题解:本题主要考查树链剖分,写了好久!
简要题意:树上两个操作,1:查询从该节点到根有多少个结点权值为0,并且把0−x上的所有节点都赋为1。2:查询该节点为根的子树总权值和,并且更新子树全部置为0
1.树链剖分:其实本题主要难点是理解题意,怎么把题目的信息转化为上面的题意。就会发现其实是树链剖分的模板改动一点,区间覆盖。
这里用了输出的优化,总和就为:ans=abs(操作前的总和-操作后的总和)。(原来我没有用,看了题解就改过来了)
代码如下:
#include<iostream>
#include<algorithm>
using namespace std;
struct N
{
int ans,lazy;
}t[800005];
struct E
{
int start,to;
}e[800005];
int siz[200005],d[200005],son[200005],h[200005];
int top[200005],rk[200005],f[200005],id[200005];
int n,m,P,num,sum;
void pushup(int p)
{
t[p].ans=t[p*2].ans+t[p*2+1].ans;
}
void build(int p,int l,int r)
{
t[p].lazy=-1;t[p].ans=0;
if(l==r)return ;
int mid=l+r>>1;
build(p*2,l,mid);
build(p*2+1,mid+1,r);
pushup(p);
return ;
}
void lai(int p,int l,int r)
{
int mid=l+r>>1;
if(t[p].lazy!=-1)
{
t[p*2].lazy=t[p*2+1].lazy=t[p].lazy;
t[p*2].ans=t[p].lazy*(mid-l+1);
t[p*2+1].ans=t[p].lazy*(r-mid);
t[p].lazy=-1;
}
return ;
}
void change(int p,int l,int r,int x,int y,int z)
{
if(x>r||y<l)return ;
if(x<=l&&y>=r)
{
t[p].ans=z*(r-l+1);
t[p].lazy=z;
return ;
}
lai(p,l,r);
int mid=l+r>>1;
if(x<=mid)change(p*2,l,mid,x,y,z);
if(y>mid)change(p*2+1,mid+1,r,x,y,z);
pushup(p);
return ;
}
void add(int start,int to)
{
e[++P].to=to;
e[P].start=h[start];
h[start]=P;
}
void dfs1(int p,int fa,int deep)
{
d[p]=deep;f[p]=fa;siz[p]=1;
for(int i=h[p];i;i=e[i].start)
{
int k=e[i].to;
if(k==fa)continue;
dfs1(k,p,deep+1);
siz[p]+=siz[k];
if(!son[p]||siz[k]>siz[son[p]])son[p]=k;
}
}
void dfs2(int p,int tp)
{
id[p]=++num;rk[num]=p;top[p]=tp;
if(!son[p])return ;
dfs2(son[p],tp);
for(int i=h[p];i;i=e[i].start)
{
int k=e[i].to;
if(k!=son[p]&&k!=f[p])
dfs2(k,k);
}
}
void update(int x,int y,int v)
{
while(top[x]!=top[y])
{
if(d[top[x]]<d[top[y]])swap(x,y);
change(1,1,n,id[top[x]],id[x],v);
x=f[top[x]];
}
if(d[x]>d[y])swap(x,y);
change(1,1,n,id[x],id[y],v);
}
int main()
{
cin>>n;
for(int i=2;i<=n;i++)
{
int x;
cin>>x;x++;
add(x,i);add(i,x);
}
dfs1(1,1,1);dfs2(1,1);
build(1,1,n);
cin>>m;
for(int i=1;i<=m;i++)
{
char a[129193];int x;
cin>>a;cin>>x;x++;
sum=t[1].ans;
if(a[0]=='i')
{
update(1,x,1);
cout<<t[1].ans-sum<<endl;
}
else if(a[0]=='u')
{
change(1,1,n,id[x],id[x]+siz[x]-1,0);
cout<<sum-t[1].ans<<endl;
}
}
return 0;
}