bzoj 4196: [Noi2015]软件包管理器

Description

 Linux用户和OSX用户一定对软件包管理器不会陌生。通过软件包管理器，你可以通过一行命令安装某一个软件包，然后软件包管理器会帮助你从软件源下载软件包，同时自动解决所有的依赖（即下载安装这个软件包的安装所依赖的其它软件包），完成所有的配置。Debian/Ubuntu使用的apt-get，Fedora/CentOS使用的yum，以及OSX下可用的homebrew都是优秀的软件包管理器。

你决定设计你自己的软件包管理器。不可避免地，你要解决软件包之间的依赖问题。如果软件包A依赖软件包B，那么安装软件包A以前，必须先安装软件包B。同时，如果想要卸载软件包B，则必须卸载软件包A。现在你已经获得了所有的软件包之间的依赖关系。而且，由于你之前的工作，除0号软件包以外，在你的管理器当中的软件包都会依赖一个且仅一个软件包，而0号软件包不依赖任何一个软件包。依赖关系不存在环（若有m(m≥2)个软件包A1,A2,A3,…,Am，其中A1依赖A2，A2依赖A3，A3依赖A4，……，Am−1依赖Am，而Am依赖A1，则称这m个软件包的依赖关系构成环），当然也不会有一个软件包依赖自己。

现在你要为你的软件包管理器写一个依赖解决程序。根据反馈，用户希望在安装和卸载某个软件包时，快速地知道这个操作实际上会改变多少个软件包的安装状态（即安装操作会安装多少个未安装的软件包，或卸载操作会卸载多少个已安装的软件包），你的任务就是实现这个部分。注意，安装一个已安装的软件包，或卸载一个未安装的软件包，都不会改变任何软件包的安装状态，即在此情况下，改变安装状态的软件包数为0。

Input

输入文件的第1行包含1个正整数n，表示软件包的总数。软件包从0开始编号。

随后一行包含n−1个整数，相邻整数之间用单个空格隔开，分别表示1,2,3,…,n−2,n−1号软件包依赖的软件包的编号。

接下来一行包含1个正整数q，表示询问的总数。

之后q行，每行1个询问。询问分为两种：

installx：表示安装软件包x

uninstallx：表示卸载软件包x

你需要维护每个软件包的安装状态，一开始所有的软件包都处于未安装状态。对于每个操作，你需要输出这步操作会改变多少个软件包的安装状态，随后应用这个操作（即改变你维护的安装状态）。

Output

输出文件包括q行。

输出文件的第i行输出1个整数，为第i步操作中改变安装状态的软件包数。

Sample Input

7
0 0 0 1 1 5
5
install 5
install 6
uninstall 1
install 4
uninstall 0

Sample Output

3
1
3
2
3

HINT

 一开始所有的软件包都处于未安装状态。

安装 5 号软件包，需要安装 0,1,5 三个软件包。

之后安装 6 号软件包，只需要安装 6 号软件包。此时安装了 0,1,5,6 四个软件包。

卸载 1 号软件包需要卸载 1,5,6 三个软件包。此时只有 0 号软件包还处于安装状态。

之后安装 4 号软件包，需要安装 1,4 两个软件包。此时 0,1,4 处在安装状态。

最后，卸载 0 号软件包会卸载所有的软件包。

n=100000

q=100000

树链剖分裸题

一开始用cin读入字符串结果T掉了。。

感觉现在写代码是慢了，当时在考场上40分钟就切掉了

0表示未安装1表示安装

首先对于安装操作，就是询问0到x上0的个数

这个直接树链剖分维护链即可

对于卸载操作，就是询问x子树中有多少个1

因为树链剖分是按照dfs序建立，因此x的子树一定是连续的一段，直接在线段树中查询即可

#include<map>
#include<queue>
#include<vector>
#include<cstdio>
#include<string>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
struct line
{
     int s,t;
     int next;
}a[400001];
int edge;
int head[200001];
inline void add(int s,int t)
{
	 a[edge].next=head[s];
     head[s]=edge;
     a[edge].s=s;
     a[edge].t=t;
}
struct tree
{
     int l,r;
     int s,tag;
}tr[800001];
inline void up(int p)
{
     tr[p].s=tr[p*2].s+tr[p*2+1].s;
}
inline void push(int p)
{
	if(tr[p].tag==-1)
		return ;
	int l=(tr[p].r-tr[p].l+1);
	tr[p*2].s=(l-l/2)*tr[p].tag;
	tr[p*2+1].s=(l/2)*tr[p].tag;
	tr[p*2].tag=tr[p].tag;
	tr[p*2+1].tag=tr[p].tag;
	tr[p].tag=-1;
}
inline void build(int p,int l,int r)
{
     tr[p].l=l;
     tr[p].r=r;
     tr[p].tag=-1;
     if(l!=r)
     {
          int mid=(l+r)/2;
          build(p*2,l,mid);
          build(p*2+1,mid+1,r);
     }
}
inline void cov(int p,int l,int r,int x)
{
     if(l<=tr[p].l&&tr[p].r<=r)
     {
          tr[p].s=(tr[p].r-tr[p].l+1)*x;
          tr[p].tag=x;
     }
     else
     {
     	  push(p);
	      int mid=(tr[p].l+tr[p].r)/2;
		  if(l<=mid)
		       cov(p*2,l,r,x);
		  if(r>mid)
		       cov(p*2+1,l,r,x);
		  up(p);
     }
}
inline int ask(int p,int l,int r,int x)
{
     if(l<=tr[p].l&&tr[p].r<=r)
     {
     	if(x==1)
     		return tr[p].s;
     	return tr[p].r-tr[p].l+1-tr[p].s;
     }
     else
     {
     	  push(p);
	      int mid=(tr[p].l+tr[p].r)/2;
	      int ans=0;
		  if(l<=mid)
		       ans+=ask(p*2,l,r,x);
		  if(r>mid)
		       ans+=ask(p*2+1,l,r,x);
		  return ans;
     }
}
int dep[200001],size[200001],son[200001],fa[200001];
int top[200001],w[200001];
int ld[200001],rd[200001];
int lson[200001]/*最大节点位置*/,mson[200001]/*最大节点值*/;
int tot;
inline void dfs1(int d)
{
     int i;
     for(i=head[d];i!=0;i=a[i].next)
     {
     	  int t=a[i].t;
          if(t!=fa[d])
          {
          	   dep[t]=dep[d]+1;
               fa[t]=d;
               dfs1(t);
               son[d]+=son[t]+1;
               if(son[t]>=mson[d])
               {
                    mson[d]=son[t];
                    lson[d]=t;
               }
          }
     }
}
inline void dfs2(int d)//先dfs重链 
{
	 ld[d]=tot;
     int i;
     for(i=head[d];i!=0;i=a[i].next)
     {
          int t=a[i].t;
          if(t==lson[d])
          {
               top[t]=top[d];
               tot++;
               w[t]=tot;
               dfs2(t);
          }
     }
     for(i=head[d];i!=0;i=a[i].next)
     {
          int t=a[i].t;
          if(t!=fa[d]&&t!=lson[d])
          {
               top[t]=t;
               tot++;
               w[t]=tot;
               dfs2(t);
          }
     }
     rd[d]=tot;
}
tree xx;
inline int askx(int s,int t,int x)//深的那个点往当前链的top跳 
{
	 int u=top[s],v=top[t];
	 int ans=0;
     while(u!=v)
     {
          if(dep[u]>dep[v])
          {
          	   ans+=ask(1,w[u],w[s],x);
               s=fa[top[s]];
          }
          else
          {
          	   ans+=ask(1,w[v],w[t],x);
               t=fa[top[t]];
          }
          u=top[s];
          v=top[t];
     }
     if(w[s]<w[t])
          ans+=ask(1,w[s],w[t],x);
     else
          ans+=ask(1,w[t],w[s],x);
     return ans;
}
inline void cover(int s,int t,int x)//深的那个点往当前链的top跳 
{
	 int u=top[s],v=top[t];
	 int ans=0;
     while(u!=v)
     {
          if(dep[u]>dep[v])
          {
          	   cov(1,w[u],w[s],x);
               s=fa[top[s]];
          }
          else
          {
          	   cov(1,w[v],w[t],x);
               t=fa[top[t]];
          }
          u=top[s];
          v=top[t];
     }
     if(w[s]<w[t])
          cov(1,w[s],w[t],x);
     else
          cov(1,w[t],w[s],x);
}
int main()
{
//	 freopen("ex_manager3.in","r",stdin);
//	 freopen("ex_manager3.ans","w",stdout);
     int n;
     scanf("%d",&n);
     int i,s,t;
     for(i=1;i<=n-1;i++)
     {
          scanf("%d",&s);
          edge++;
          add(s+1,i+1);
     }
     dep[1]=1;
     dfs1(1);
     top[1]=1;
     tot++;
     w[1]=tot;
     dfs2(1);
     build(1,1,tot);
     int m;
     scanf("%d",&m);
   //  string x;
   	 char x[11];
     int xx;
     for(i=1;i<=m;i++)
     {
     	//cin>>x;
     	scanf("%s",x);
     	if(x[0]=='i')
     	{
     		scanf("%d",&xx);
     		xx++;
     		printf("%d\n",askx(1,xx,0));
     		cover(1,xx,1);
     	}
     	else
     	{
     		scanf("%d",&xx);
     		xx++;
     		printf("%d\n",ask(1,ld[xx],rd[xx],1));
     		cov(1,ld[xx],rd[xx],0);
     	}
     }
     return 0;
}