题目背景
Linux 用户和 OSX 用户一定对软件包管理器不会陌生。通过软件包管理器,你可以通过一行命令安装某一个软件包,然后软件包管理器会帮助你从软件源下载软件包,同时自动解决所有的依赖(即下载安装这个软件包的安装所依赖的其它软件包),完成所有的配置。Debian/Ubuntu 使用的 apt-get,Fedora/CentOS 使用的 yum,以及 OSX 下可用的 homebrew 都是优秀的软件包管理器。
题目描述
你决定设计你自己的软件包管理器。不可避免地,你要解决软件包之间的依赖问题。如果软件包 a 依赖软件包 b,那么安装软件包 a 以前,必须先安装软件包 b。同时,如果想要卸载软件包 b,则必须卸载软件包 a。
现在你已经获得了所有的软件包之间的依赖关系。而且,由于你之前的工作,除 00 号软件包以外,在你的管理器当中的软件包都会依赖一个且仅一个软件包,而 00 号软件包不依赖任何一个软件包。且依赖关系不存在环。
现在你要为你的软件包管理器写一个依赖解决程序。根据反馈,用户希望在安装和卸载某个软件包时,快速地知道这个操作实际上会改变多少个软件包的安装状态(即安装操作会安装多少个未安装的软件包,或卸载操作会卸载多少个已安装的软件包),你的任务就是实现这个部分。
注意,安装一个已安装的软件包,或卸载一个未安装的软件包,都不会改变任何软件包的安装状态,即在此情况下,改变安装状态的软件包数为 0。
输入格式
第一行一个正整数 n,表示软件包个数,从 0 开始编号。
第二行有 n−1 个整数,第 i个表示 i号软件包依赖的软件包编号。
然后一行一个正整数 q,表示操作个数,格式如下:
install x 表示安装 x 号软件包
uninstall x 表示卸载 x 号软件包
一开始所有软件包都是未安装的。
对于每个操作,你需要输出这步操作会改变多少个软件包的安装状态,随后应用这个操作(即改变你维护的安装状态)。
输出格式
输出 q 行,每行一个整数,表示每次询问的答案。
输入输出样例
输入 #1
7
0 0 0 1 1 5
5
install 5
install 6
uninstall 1
install 4
uninstall 0
输出 #1
3
1
3
2
3
输入 #2
10
0 1 2 1 3 0 0 3 2
10
install 0
install 3
uninstall 2
install 7
install 5
install 9
uninstall 9
install 4
install 1
install 9
输出 #2
1
3
2
1
3
1
1
1
0
1
树链剖分模板题
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
#define fp(i,a,b) for(int i=a;i<=b;i++)
int e[N],ne[N],h[N],idx;
int tid[N],top[N],son[N],father[N],deep[N],size[N],tidnum;
int n,q;
char s[15];
void add(int x,int y)
{
e[idx]=y;
ne[idx]=h[x];
h[x]=idx++;
}
struct node
{
int sum,lazy;
} t[N*4];
void dfs1(int x,int fa,int depth)
{
size[x]=1;
father[x]=fa;
deep[x]=depth;
for(int i=h[x]; ~i; i=ne[i])
{
int j=e[i];
if(j==fa) continue;
dfs1(j,x,depth+1);
size[x]+=size[j];
if(!son[x]||size[j]>size[son[x]]) son[x]=j;
}
}
void dfs2(int x,int tp)
{
tid[x]=++tidnum;
top[x]=tp;
if(!son[x]) return;
dfs2(son[x],tp);
for(int i=h[x]; ~i; i=ne[i])
{
int j=e[i];
if(j!=son[x]&&j!=father[x])
dfs2(j,j);
}
}
void pushup(int rt)
{
t[rt].sum=t[rt<<1].sum+t[rt<<1|1].sum;
}
void pushdown(int rt,int l,int r)
{
if(t[rt].lazy!=-1)
{
int mid=(l+r)>>1;
t[rt<<1].sum=t[rt].lazy*(mid-l+1);
t[rt<<1].lazy=t[rt].lazy;
t[rt<<1|1].sum=t[rt].lazy*(r-mid);
t[rt<<1|1].lazy=t[rt].lazy;
t[rt].lazy=-1;
}
}
void build(int rt,int l,int r)
{
t[rt].lazy=-1;
if(l==r)
{
t[rt].sum=0;
return ;
}
int mid=(l+r)>>1;
build(rt<<1,l,mid);
build(rt<<1|1,mid+1,r);
pushup(rt);
}
void update(int rt,int l,int r,int L,int R,int val)
{
if(L<=l && R>=r)
{
t[rt].sum=val*(r-l+1);
t[rt].lazy=val;
return;
}
pushdown(rt,l,r);
int mid=(l+r)>>1;
if(L<=mid) update(rt<<1,l,mid,L,R,val);
if(R>mid)update(rt<<1|1,mid+1,r,L,R,val);
pushup(rt);
}
void change(int u,int v,int val)
{
while(top[u]!=top[v])
{
if(deep[top[u]]<deep[top[v]]) swap(u,v);
update(1,1,n,tid[top[u]],tid[u],val);
u=father[top[u]];
}
if(deep[u]>deep[v]) std::swap(u,v);
update(1,1,n,tid[u],tid[v],val);
return;
}
int main()
{
memset(h,-1,sizeof h);
scanf("%d",&n);
fp(i,2,n)
{
int x;
scanf("%d",&x);
x++;
add(x,i);
}
dfs1(1,1,1);
dfs2(1,1);
scanf("%d",&q);
fp(i,1,q)
{
scanf("%s",s);
int x;
scanf("%d",&x);
x++;
int t1=t[1].sum;
if(s[0]=='i')
{
change(1,x,1);
int t2=t[1].sum;
printf("%d\n",abs(t2-t1));
}
else if(s[0]=='u')
{
update(1,1,n,tid[x],tid[x]+size[x]-1,0);
int t2=t[1].sum;
printf("%d\n",abs(t2-t1));
}
}
return 0;
}