传送门
链剖练习题,思路很好想。
刚开始的时候出了个小bug,让我现在十分怀疑自己写数据结构的一次成功率。
#include
#include
#include
using namespace std;
const int max_n=1e5+5;
const int max_e=max_n*2;
const int max_tree=max_n*5;
int n,q,x,N,ans;
char s[20];
int tot,point[max_n],next[max_e],v[max_e];
int size[max_n],h[max_n],father[max_n],son[max_n];
int top[max_n],in[max_n],out[max_n];
int sum[max_tree],delta[max_tree];
inline void addedge(int x,int y){
++tot; next[tot]=point[x]; point[x]=tot; v[tot]=y;
++tot; next[tot]=point[y]; point[y]=tot; v[tot]=x;
}
inline void dfs_1(int x,int fa,int dep){
size[x]=1; h[x]=dep; father[x]=fa;
int maxson=0;
for (int i=point[x];i;i=next[i])
if (v[i]!=fa){
dfs_1(v[i],x,dep+1);
size[x]+=size[v[i]];
if (maxsoninline void dfs_2(int x,int fa){
if (son[fa]!=x) top[x]=x;
else top[x]=top[fa];
in[x]=++N;
if (son[x]) dfs_2(son[x],x);
for (int i=point[x];i;i=next[i])
if (v[i]!=fa&&v[i]!=son[x])
dfs_2(v[i],x);
out[x]=N;
}
inline void update(int now){
sum[now]=sum[now<<1]+sum[now<<1|1];
}
inline void pushdown(int now,int l,int r,int mid){
if (delta[now]>=0){
sum[now<<1]=delta[now]*(mid-l+1);
delta[now<<1]=delta[now];
sum[now<<1|1]=delta[now]*(r-mid);
delta[now<<1|1]=delta[now];
delta[now]=-1;
}
}
inline void interval_change(int now,int l,int r,int lrange,int rrange,int v){
int mid=(l+r)>>1;
if (lrange<=l&&r<=rrange){
sum[now]=(r-l+1)*v;
delta[now]=v;
return;
}
pushdown(now,l,r,mid);
if (lrange<=mid)
interval_change(now<<1,l,mid,lrange,rrange,v);
if (mid+1<=rrange)
interval_change(now<<1|1,mid+1,r,lrange,rrange,v);
update(now);
}
inline int query(int now,int l,int r,int lrange,int rrange){
int mid=(l+r)>>1,ans=0;
if (lrange<=l&&r<=rrange) return sum[now];
pushdown(now,l,r,mid);
if (lrange<=mid)
ans+=query(now<<1,l,mid,lrange,rrange);
if (mid+1<=rrange)
ans+=query(now<<1|1,mid+1,r,lrange,rrange);
return ans;
}
inline void QUERY(int u,int t){
int f1=top[u],f2=top[t],ans=0;
while (f1!=f2){
if (h[f1]1,1,N,in[f1],in[u]);
u=father[f1];
f1=top[u];
}
if (in[u]>in[t]) swap(u,t);
ans+=query(1,1,N,in[u],in[t]);
int cnt=h[x];
printf("%d\n",cnt-ans);
}
inline void CHANGE(int u,int t){
int f1=top[u],f2=top[t],ans=0;
while (f1!=f2){
if (h[f1]1,1,N,in[f1],in[u],1);
u=father[f1];
f1=top[u];
}
if (in[u]>in[t]) swap(u,t);
interval_change(1,1,N,in[u],in[t],1);
}
inline void INSTALL(int u,int t){
QUERY(u,t);
CHANGE(u,t);
}
inline void UNINSTALL(int x){
ans=query(1,1,N,in[x],out[x]);
printf("%d\n",ans);
interval_change(1,1,N,in[x],out[x],0);
}
int main(){
// freopen("input.txt","r",stdin);
scanf("%d",&n);
for (int i=1;iscanf("%d",&x);
++x;
addedge(x,i+1);
}
dfs_1(1,0,1);
dfs_2(1,0);
//0未安装 1已安装
memset(sum,0,sizeof(sum));
memset(delta,128,sizeof(delta));
scanf("%d",&q);
for (int i=1;i<=q;++i){
scanf("%s",s);
scanf("%d",&x);
++x;
if (s[0]=='i')
INSTALL(1,x);
else
UNINSTALL(x);
}
}
注意线段树中的初始状态(尤其delta)不能与线段树中要出现的状态重合(像本题中的0)。线段树是要直接修改还是要加减要注意。