注意点:
- 动态开点线段树如果写区间覆盖问题一定要保证子结点存在.
- 数据点里结点从0开始编号,平移加一即可.
- 区间覆盖线段树需要写pushdown和pushup,query和insert也有一定的改动,但整体上难度降低.
#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
const int MAXN=1000010,MAXM=1000010;
struct Node{
int ls,rs;
int tag,sum;
}tr[MAXN];
int root=0,nodeCnt=0;
void init(int now){
if(!tr[now].ls)tr[now].ls=++nodeCnt;
if(!tr[now].rs)tr[now].rs=++nodeCnt;
}
void pushup(int now){
init(now);
tr[now].sum=tr[tr[now].ls].sum+tr[tr[now].rs].sum;
}
void pushdown(int now,int l,int r){
init(now);
int mid=(l+r)>>1;
tr[tr[now].ls].tag=tr[now].tag;
tr[tr[now].rs].tag=tr[now].tag;
if(tr[now].tag==1)tr[tr[now].ls].sum=tr[tr[now].rs].sum=0;
else if(tr[now].tag==2)tr[tr[now].ls].sum=mid-l+1,tr[tr[now].rs].sum=r-mid;
tr[now].tag=0;
}
void insert(int &now,int l,int r,int ll,int rr,int x){
if(!now)now=++nodeCnt;
init(now);
if(ll<=l&&rr>=r){
tr[now].tag=x;
if(x==1)tr[now].sum=0;
else if(x==2)tr[now].sum=r-l+1;
return;
}
int mid=(l+r)>>1;
if(tr[now].tag)pushdown(now,l,r);
if(ll<=mid)insert(tr[now].ls,l,mid,ll,rr,x);
if(rr>mid)insert(tr[now].rs,mid+1,r,ll,rr,x);
pushup(now);
}
int query(int now,int l,int r,int ll,int rr){
if(ll<=l&&rr>=r)return tr[now].sum;
int mid=(l+r)>>1;
init(now);
if(tr[now].tag){
pushdown(now,l,r);
}
int ans=0;
if(ll<=mid)ans+=query(tr[now].ls,l,mid,ll,rr);
if(rr>mid)ans+=query(tr[now].rs,mid+1,r,ll,rr);
return ans;
}
struct Edge{
int from,to,nxt;
}e[MAXM];
int head[MAXN],edgeCnt=1;
void addEdge(int u,int v){
e[++edgeCnt].from=u;
e[edgeCnt].to=v;
e[edgeCnt].nxt=head[u];
head[u]=edgeCnt;
}
int dep[MAXN],fa[MAXN],son[MAXN],siz[MAXN];
void dfs1(int x){
dep[x]=dep[fa[x]]+1;
siz[x]=1;
for(int i=head[x];i;i=e[i].nxt){
int nowV=e[i].to;
if(!dep[nowV]){
fa[nowV]=x;
dfs1(nowV);
siz[x]+=siz[nowV];
if(siz[nowV]>siz[son[x]])son[x]=nowV;
}
}
}
int id[MAXN],idCnt=0;
int initValue[MAXN],top[MAXN];
void dfs2(int x){
id[x]=++idCnt;
if(x==son[fa[x]])top[x]=top[fa[x]];
else top[x]=x;
if(son[x])dfs2(son[x]);
for(int i=head[x];i;i=e[i].nxt){
int nowV=e[i].to;
if(!id[nowV])
dfs2(nowV);
}
}
void modify(int u,int v,int x){
while(top[u]!=top[v]){
if(dep[top[u]]>dep[top[v]])swap(u,v);
insert(root,1,MAXN,id[top[v]],id[v],x);
v=fa[top[v]];
}
insert(root,1,MAXN,min(id[u],id[v]),max(id[u],id[v]),x);
}
int sum(int u,int v){
int ans=0;
while(top[u]!=top[v]){
if(dep[top[u]]>dep[top[v]])swap(u,v);
ans+=id[v]-id[top[v]]+1-query(root,1,MAXN,id[top[v]],id[v]);
v=fa[top[v]];
}
if(dep[u]<dep[v])swap(u,v);
ans+=id[u]-id[v]+1-query(root,1,MAXN,id[v],id[u]);
return ans;
}
int main(){
int n;
scanf("%d",&n);
for(int i=2;i<=n;i++){
int v;
scanf("%d",&v);
v=v+1;
addEdge(v,i);
addEdge(i,v);
}
dfs1(1);
dfs2(1);
int q;
scanf("%d",&q);
string str;
int x;
for(int i=1;i<=q;i++){
cin>>str>>x;
x=x+1;
if(str[0]=='i'){
cout<<sum(1,x)<<endl;
modify(1,x,2);
}else if(str[0]=='u'){
cout<<query(root,1,MAXN,id[x],id[x]+siz[x]-1)<<endl;
insert(root,1,MAXN,id[x],id[x]+siz[x]-1,1);
}
}
return 0;
}