标签:getchar int style class ddd php tin memset fine
install x-> 询问根节点到x路径上0的个数,然后全变1
uninstall x-> 询问x子树(包括x)中1的个数,然后全边0
#include <cstdio> #include <algorithm> #include <cstring> #define MID int mid=(l+r)>>1,ls=id<<1,rs=id<<1|1 #define len (r-l+1) #define N 100010 using namespace std; struct info{int to,nex;}e[N*2]; int n,A[N],tot,head[N],T[N*4],tag[N*4]; int dep[N],fa[N],sz[N],son[N]; int cnt,tp[N],tw[N],tid[N]; inline int read(){ int x=0,f=1;char ch=getchar(); while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();} while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();} return x*f; } void dfs(int u,int pre){ sz[u]=1; for(int i=head[u],mx=0;i;i=e[i].nex){ int v=e[i].to; if(v==pre) continue; dep[v]=dep[u]+1; fa[v]=u; dfs(v,u); sz[u]+=sz[v]; if(sz[v]>mx) son[u]=v,mx=sz[v]; } } inline void Link(int u,int v){ e[++tot].nex=head[u];e[tot].to=v;head[u]=tot; } void dddfs(int u,int top){ tp[u]=top; tid[u]=++cnt; tw[cnt]=A[u]; if(!son[u]) return; dddfs(son[u],top); for(int i=head[u];i;i=e[i].nex){ int v=e[i].to; if(v!=son[u]&&v!=fa[u]) dddfs(v,v); } } inline void Init(){ memset(tag,-1,sizeof(tag)); n=read(); for(int i=1;i<n;++i){ int u=i,v=read(); Link(u,v),Link(v,u); } dfs(0,0); dddfs(0,0); } void pushdown(int l,int r,int id){ MID; tag[ls]=tag[rs]=tag[id]; T[ls]=tag[id]*(mid-l+1); T[rs]=tag[id]*(r-mid); tag[id]=-1; } int query(int l,int r,int id,int L,int R){ if(L<=l&&r<=R) return T[id]; if(tag[id]!=-1) pushdown(l,r,id); MID; int res=0; if(L<=mid) res+=query(l,mid,ls,L,R); if(R>mid) res+=query(mid+1,r,rs,L,R); return res; } inline int qRange(int u,int v){ int res=0; while(tp[u]!=tp[v]){ if(dep[tp[u]]<dep[tp[v]]) swap(u,v); res+=query(1,n,1,tid[tp[u]],tid[u]); u=fa[tp[u]]; } if(dep[u]>dep[v]) swap(u,v); res+=query(1,n,1,tid[u],tid[v]); return res; } void update(int l,int r,int id,int L,int R,int f){ if(L<=l&&r<=R){ tag[id]=f;T[id]=len*f; return; } if(tag[id]!=-1) pushdown(l,r,id); MID; if(tag[id]) if(L<=mid) update(l,mid,ls,L,R,f); if(R>mid) update(mid+1,r,rs,L,R,f); T[id]=T[ls]+T[rs]; } inline void updRange(int u,int v,int f){ while(tp[u]!=tp[v]){ if(dep[tp[u]]<dep[tp[v]]) swap(u,v); update(1,n,1,tid[tp[u]],tid[u],f); u=fa[tp[u]]; } if(dep[u]>dep[v]) swap(u,v); update(1,n,1,tid[u],tid[v],f); } int ins(int u){ int res=dep[u]+1-qRange(0,u); updRange(0,u,1); return res; } int uni(int u){ int res=query(1,n,1,tid[u],tid[u]+sz[u]-1); update(1,n,1,tid[u],tid[u]+sz[u]-1,0); return res; } inline void solve(){ int m=read(),u; char s[20]; while(m--){ scanf("%s%d\n",s,&u); if(s[0]==‘i‘) printf("%d\n",ins(u)); else printf("%d\n",uni(u)); } } int main(){Init();solve();}
标签:getchar int style class ddd php tin memset fine
原文地址:https://www.cnblogs.com/void-f/p/9016225.html