树链剖分
看懂题目后就是比较裸的树剖了。安装就是把x到根节点都改为1,卸载就是把x的子树都改为0。
注意询问的内容。
论10s时限跑了9.9s的快感。
代码:
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define F inline
using namespace std;
struct tree{ int l,r,x,f; }t[N<<2];
struct edge{ int nxt,to; }ed[N<<1];
int n,m,k,p,h[N],fa[N],sz[N],to[N],tp[N],L[N],R[N];
F char readc(){
static char buf[100000],*l=buf,*r=buf;
if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
return l==r?EOF:*l++;
}
F int _read(){
int x=0; char ch=readc();
while (!isdigit(ch)&&ch!='i'&&ch!='u') ch=readc();
if (ch=='i'||ch=='u'){
char s=ch;
for (int i=1;i<=7+2*(s=='u');i++) ch=readc();
return s;
}
while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
return x;
}
F void writec(int x){ if (x>9) writec(x/10); putchar(x%10+48); }
F void _write(int x){ writec(x),puts(""); }
#define addedge(x,y) ed[++k]=(edge){h[x],y},h[x]=k
void dfs1(int x){
sz[x]=1;
for (int i=h[x],v;i;i=ed[i].nxt){
dfs1(v=ed[i].to),sz[x]+=sz[v];
if (sz[v]>sz[to[x]]) to[x]=v;
}
}
void dfs2(int x){
L[x]=++p;
if (to[x]) tp[to[x]]=tp[x],dfs2(to[x]);
for (int i=h[x],v;i;i=ed[i].nxt)
if ((v=ed[i].to)!=to[x])
tp[v]=v,dfs2(v);
R[x]=p;
}
void build(int x,int l,int r){
t[x].l=l,t[x].r=r,t[x].f=-1;
if (l==r) return; int mid=l+r>>1;
build(x<<1,l,mid),build(x<<1|1,mid+1,r);
}
F void pshd(int x){
t[x<<1].f=t[x<<1|1].f=t[x].f,t[x].f=-1;
t[x<<1].x=(t[x<<1].r-t[x<<1].l+1)*t[x<<1].f;
t[x<<1|1].x=(t[x<<1|1].r-t[x<<1|1].l+1)*t[x<<1|1].f;
}
void mdfy(int x,int l,int r,int w){
if (t[x].l>r||t[x].r<l) return;
if (t[x].l>=l&&t[x].r<=r){
t[x].x=(t[x].r-t[x].l+1)*w;
t[x].f=w; return;
}
if (~t[x].f) pshd(x);
mdfy(x<<1,l,r,w),mdfy(x<<1|1,l,r,w);
t[x].x=t[x<<1].x+t[x<<1|1].x;
}
int srch(int x,int l,int r){
if (t[x].l>r||t[x].r<l) return 0;
if (t[x].l>=l&&t[x].r<=r) return t[x].x;
if (~t[x].f) pshd(x);
return srch(x<<1,l,r)+srch(x<<1|1,l,r);
}
void nsrt(int x){
while (tp[x]!=1)
mdfy(1,L[tp[x]],L[x],1),x=fa[tp[x]];
mdfy(1,1,L[x],1);
}
int find(int x){
int ans=0;
while (tp[x]!=1)
ans+=srch(1,L[tp[x]],L[x]),x=fa[tp[x]];
return ans+srch(1,1,L[x]);
}
int main(){
n=_read();
for (int i=2;i<=n;i++)
fa[i]=_read()+1,addedge(fa[i],i);
dfs1(1),tp[1]=1,dfs2(1),build(1,1,n);
for (m=_read();m;m--){
int f=_read(),x=_read()+1,y;
if (f=='i') y=find(x),nsrt(x),_write(find(x)-y);
else{
y=srch(1,L[x],R[x]),mdfy(1,L[x],R[x],0);
_write(y-srch(1,L[x],R[x]));
}
}
return 0;
}