题目描述 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,……,A[m-1]依赖Am,而Am依赖A1,则称这m个软件包的依赖关系构成环),当然也不会有一个软件包依赖自己。 现在你要为你的软件包管理器写一个依赖解决程序。根据反馈,用户希望在安装和卸载某个软件包时,快速地知道这个操作实际上会改变多少个软件包的安装状态(即安装操作会安装多少个未安装的软件包,或卸载操作会卸载多少个已安装的软件包),你的任务就是实现这个部分。注意,安装一个已安装的软件包,或卸载一个未安装的软件包,都不会改变任何软件包的安装状态,即在此情况下,改变安装状态的软件包数为0。 输入格式 从文件manager.in中读入数据。 输入文件的第1行包含1个整数n,表示软件包的总数。软件包从0开始编号。 随后一行包含n−1个整数,相邻整数之间用单个空格隔开,分别表示1,2,3,⋯,n−2,n−1号软件包依赖的软件包的编号。 接下来一行包含1个整数q,表示询问的总数。之后q行,每行1个询问。询问分为两种: install x:表示安装软件包x uninstall x:表示卸载软件包x 你需要维护每个软件包的安装状态,一开始所有的软件包都处于未安装状态。 对于每个操作,你需要输出这步操作会改变多少个软件包的安装状态,随后应用这个操作(即改变你维护的安装状态)。 输出格式 输出到文件manager.out中。 输出文件包括q行。 输出文件的第i行输出1个整数,为第i步操作中改变安装状态的软件包数。
输入输出样例 输入 #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
说明/提示
解释:如果存在我们记为1,不存在我们记为0,这样我们就转行成树上求和问题了,直接上树链剖分模板
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #define lson rt<<1 #define rson rt<<1|1 #define ll long long const ll N=100003; const ll inf=-100000009; using namespace std; ll n,idx,dfn[N],seq[N],fa[N],dep[N],top[N],sz[N],hvy[N]; ll dp[N]={0}; ll tree[8*N]={0},val[N]={0}; ll a[N]={0}; ll lazy[8*N]={0}; ll tot,lnk[N],ter[4*N],nxt[4*N]; void add(ll u,ll v){ ter[++tot]=v,nxt[tot]=lnk[u],lnk[u]=tot; } void dfs1(ll u,ll f){ fa[u]=f,dep[u]=dep[f]+1,sz[u]=1,hvy[u]=top[u]=0; for(ll i=lnk[u];i;i=nxt[i]){ ll v=ter[i]; if(v==f) continue; dp[v]=dp[u]+1; dfs1(v,u); sz[u]+=sz[v]; if(sz[hvy[u]]<sz[v]) hvy[u]=v; } } void dfs2(ll u,ll tp) { dfn[u]=++idx,seq[idx]=u,top[u]=tp; if(!hvy[u]) return; dfs2(hvy[u],tp); for(ll i=lnk[u];i;i=nxt[i]) { ll v=ter[i]; if(v==fa[u]||v==hvy[u]) continue; dfs2(v,v); } } void pushup(ll rt) { tree[rt]=tree[lson]+tree[rson]; } void pushdown(ll rt,ll l,ll r){ ll mid=(l+r)/2; if(lazy[rt]==-1) return; tree[lson]=lazy[rt]*(mid-l+1); tree[rson]=lazy[rt]*(r-mid); lazy[lson]=lazy[rt]; lazy[rson]=lazy[rt]; lazy[rt]=-1; return; } void build(ll rt,ll l,ll r) { if(l==r) { tree[rt]=val[seq[l]]; lazy[rt]=-1; return; } ll mid=(l+r)>>1; build(lson,l,mid); build(rson,mid+1,r); pushup(rt); } void modify(ll L,ll R,ll rt,ll l,ll r,ll val) { pushdown(rt,l,r); if(L<=l&&r<=R){ tree[rt]=val*(r-l+1); lazy[rt]=val; return; } ll mid=(l+r)>>1; if(L<=mid) modify(L,R,lson,l,mid,val); if(R>mid) modify(L,R,rson,mid+1,r,val); pushup(rt); } void chainmodify(ll u,ll v,ll val) { for(ll fu=top[u],fv=top[v];fu^fv;u=fa[fu],fu=top[u]){ if(dep[fu]<dep[fv]) swap(u,v),swap(fu,fv); modify(dfn[fu],dfn[u],1,1,n,val); } if(dep[u]>dep[v]) swap(u,v); modify(dfn[u],dfn[v],1,1,n,val); return ; } ll querySUM(ll x,ll y,ll rt,ll l,ll r){ pushdown(rt,l,r); if(x>y) return 0; if(x<=l&&r<=y) return tree[rt]; ll mid=(l+r)>>1; ll res=0; if(x<=mid) res+=querySUM(x,y,lson,l,mid); if(mid<y) res+=querySUM(x,y,rson,mid+1,r); return res; } ll chainQuerySUM(ll u,ll v){ ll res=0; for(ll fu=top[u],fv=top[v];fu^fv;u=fa[fu],fu=top[u]){ if(dep[fu]<dep[fv]) swap(u,v),swap(fu,fv); res+=querySUM(dfn[fu],dfn[u],1,1,n); } if(dep[u]>dep[v]) swap(u,v); res+=querySUM(dfn[u],dfn[v],1,1,n); return res; } int main(){ scanf("%lld",&n); for(int i=2;i<=n;i++) scanf("%lld",&a[i]); for(int i=2;i<=n;++i){ add(i,a[i]+1),add(a[i]+1,i); } dp[1]=1; dfs1(1,0),dfs2(1,1),build(1,1,n); int q=0;scanf("%d",&q); while(q--){ char cmd[123];int x;scanf("%s%d",cmd,&x);x++; ll ret=0; if(cmd[0]=='i'){ ret=dp[x]-chainQuerySUM(1,x); chainmodify(1,x,1); }else{ ret=querySUM(dfn[x],dfn[x]+sz[x]-1,1,1,n); modify(dfn[x],dfn[x]+sz[x]-1,1,1,n,0); } cout<<ret<<endl; } return 0; }