BZOJ 4196 [Noi2015]软件包管理器

来源：互联网发布：剑桥秦汉史知乎编辑：程序博客网时间：2024/06/05 21:14

题目传送门：2333333

还是裸的树剖题。

每个点0=没下载，1=已下载

下载就是查找这个点到零的路径上有几个0，再把整条路修改成1。

卸载就是查找这个点子树中有几个1，再把整棵子树改为0。

再发现一颗子树在dfs序里面是连续的一串（自己脑补），

所以就变成了线段树区间赋值+求和，随便就过了。

#include <cstdio>#include <algorithm>#include <cstring>#include <iostream>#define rep(j,k,l) for (int j=k;j<=l;j++)#define N 100005using namespace std;struct _233{int l,r,sum,flag;}tr[N*5];int n,T,cnt,ne[N*2],to[N*2],st[N];int size[N],fa[N],deep[N],son[N],top[N],dfn[N],id[N];void add(int k,int l,int p){to[p]=l;ne[p]=st[k];st[k]=p;}void dfs1(int rt,int dad){size[rt]=1;fa[rt]=dad;deep[rt]=deep[dad]+1;son[rt]=0;int _=0;for (int i=st[rt];i!=0;i=ne[i])if (to[i]!=dad){dfs1(to[i],rt);size[rt]+=size[to[i]];if (size[to[i]]>_){_=size[to[i]];son[rt]=to[i];}}}void dfs2(int rt,bool qaz){if (qaz) top[rt]=top[fa[rt]];else top[rt]=rt;dfn[++cnt]=rt;id[rt]=cnt;if (son[rt]!=0) dfs2(son[rt],1);for (int i=st[rt];i!=0;i=ne[i])if (to[i]!=fa[rt]&&to[i]!=son[rt])dfs2(to[i],0);}void stree(int k,int l,int r){if (l==r) return;tr[k].l=++cnt;tr[k].r=++cnt;stree(tr[k].l,l,(l+r)/2);stree(tr[k].r,(l+r)/2+1,r);}int check(int k,int l,int r,int o,int p,int cl){if (o>r||p<l) return 0;if (tr[k].flag>0){if (tr[k].l>0) tr[tr[k].l].flag=tr[k].flag,tr[tr[k].l].sum=((r+l)/2-l+1)*(tr[k].flag%2);if (tr[k].r>0) tr[tr[k].r].flag=tr[k].flag,tr[tr[k].r].sum=(r-(r+l)/2)*(tr[k].flag%2);tr[k].flag=0;}if (o<=l&&r<=p){int qwe=tr[k].sum;tr[k].sum=(r-l+1)*(cl%2);tr[k].flag=cl;return qwe;}int ans=check(tr[k].l,l,(l+r)/2,o,p,cl)+check(tr[k].r,(l+r)/2+1,r,o,p,cl);tr[k].sum=tr[tr[k].l].sum+tr[tr[k].r].sum;return ans;}int install(int x,int y){int ans=0;for (;top[x]!=top[y];x=fa[top[x]]){if (deep[top[x]]<deep[top[y]]) swap(x,y);ans+=check(1,1,n,id[top[x]],id[x],1);}ans+=check(1,1,n,min(id[x],id[y]),max(id[x],id[y]),1);return ans;}int uninstall(int k){return check(1,1,n,id[k],id[k]+size[k]-1,2);}int main(){scanf("%d",&n);rep(i,2,n){int k;scanf("%d",&k);add(i,k+1,2*i-3);add(k+1,i,2*i-2);}dfs1(1,0);dfs2(1,0);cnt=1;stree(1,1,n);scanf("%d",&T);while (T--){char ch=getchar();while (ch<'a'||ch>'z') ch=getchar();if (ch=='i'){rep(i,1,6) ch=getchar();int k;scanf("%d",&k);k++;printf("%d\n",deep[k]-install(1,k));}else{rep(i,1,8) ch=getchar();int k;scanf("%d",&k);k++;printf("%d\n",uninstall(k));}}system("pause");}

0 0