8说了上代码

给定一棵树,两种操作
a x:x->root路径上的点权值置1
b x: 把x的子树所有结点权值置0
树上的区间更新操作,显然是要维护dfs
第一个操作暴力显然是t,用树剖把复杂度降到log^2n级
线段树上:区间置1|0
#include<bits/stdc++.h>
using namespace std;
#define maxn 100005
struct Edge{int to,nxt;}edge[maxn<<];
int head[maxn],tot,n,q; int f[maxn],son[maxn],size[maxn],d[maxn];
void dfs1(int x,int pre,int deep){
size[x]=;d[x]=deep;f[x]=pre;
for(int i=head[x];i!=-;i=edge[i].nxt){
int y=edge[i].to;
if(y==pre)continue;
dfs1(y,x,deep+);
size[x]+=size[y];
if(size[y]>size[son[x]])son[x]=y;
}
}
int top[maxn],id[maxn],rk[maxn],cnt;
void dfs2(int x,int tp){
top[x]=tp;id[x]=++cnt;rk[cnt]=x;
if(son[x])dfs2(son[x],tp);
for(int i=head[x];i!=-;i=edge[i].nxt){
int y=edge[i].to;
if(y!=son[x] && y!=f[x])dfs2(y,y);
}
} #define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int sum[maxn<<],lazy[maxn<<];
void pushup(int rt){sum[rt]=sum[rt<<]+sum[rt<<|];}
void pushdown(int l,int r,int rt){
if(lazy[rt]<)return;
if(lazy[rt]==){
lazy[rt<<]=lazy[rt<<|]=;
sum[rt<<]=sum[rt<<|]=;
lazy[rt]=-;
}
if(lazy[rt]==){
int m=l+r>>;
lazy[rt<<]=lazy[rt<<|]=;
sum[rt<<]=m-l+;sum[rt<<|]=r-m;
lazy[rt]=-;
}
}
void set0(int L,int R,int l,int r,int rt){
if(L<=l && R>=r){lazy[rt]=;sum[rt]=;return;}
pushdown(l,r,rt);
int m=l+r>>;
if(L<=m)set0(L,R,lson);
if(R>m)set0(L,R,rson);
pushup(rt);
}
void set1(int L,int R,int l,int r,int rt){
if(L<=l && R>=r){lazy[rt]=;sum[rt]=r-l+;return;}
pushdown(l,r,rt);
int m=l+r>>;
if(L<=m)set1(L,R,lson);
if(R>m)set1(L,R,rson);
pushup(rt);
}
int query(int L,int R,int l,int r,int rt){
if(L<=l &&R >=r)return sum[rt];
int m=l+r>>,res=;
pushdown(l,r,rt);
if(L<=m)res+=query(L,R,lson);
if(R>m)res+=query(L,R,rson);
return res;
} void Set1(int x,int y){
while(top[x]!=top[y]){
if(d[top[x]]<d[top[y]])swap(x,y);
set1(id[top[x]],id[x],,n,);
x=f[top[x]];
}
if(id[x]>id[y])swap(x,y);
set1(id[x],id[y],,n,);
}
int Query(int x,int y){
int res=;
while(top[x]!=top[y]){
if(d[top[x]]<d[top[y]])swap(x,y);
res+=query(id[top[x]],id[x],,n,);
x=f[top[x]];
}
if(id[x]>id[y])swap(x,y);
return res+query(id[x],id[y],,n,);
} void init(){
memset(head,-,sizeof head);
memset(lazy,-,sizeof lazy);
tot=;
}
void addedge(int u,int v){
edge[tot].to=v;edge[tot].nxt=head[u];head[u]=tot++;
} int main(){
init();
cin>>n;
for(int u=;u<=n;u++){
int fa;scanf("%d",&fa);fa++;
addedge(fa,u);addedge(u,fa);
}
cnt=;dfs1(,,);dfs2(,); cin>>q;char op[];int x;
while(q--){
scanf("%s %d",op,&x);++x;
if(op[]=='i'){
int tmp=Query(x,);
cout<<d[x]-tmp<<'\n';
Set1(x,);
}
if(op[]=='u'){
cout<<query(id[x],id[x]+size[x]-,,n,)<<'\n';
set0(id[x],id[x]+size[x]-,,n,);
}
}
}
04-20 11:40