改了几个小时终于改完了
题目描述
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
一道十分常规的树链剖分+dfs序配合线段树的题目,但是细节较多:
1、pushdown时千万不要把右子树多乘1(即不要算上mid处)。
2、方便起见,我们可以将结点编号+1
3、修改时千万注意不要多修改,若install,则把[x,y]修改为1,维护sum即可,若uninstall,则将[x,y]修改为0,同样维护sum
时间复杂度应该是O(q*log^2(n)) 蒟蒻不太会算
代码如下:
#include <bits/stdc++.h>
using namespace std;
#define maxn 100005
int head[maxn],deep[maxn],f[maxn],son[maxn],size[maxn],top[maxn],in[maxn],out[maxn];
int n,cnt=0,tim=0;
struct node
{
int to,next;
}e[maxn*2];
struct Node
{
int l,r,sum,lazy,add;
}tree[maxn*4];
inline int read()
{
int w=1,s=0; char ch=getchar();
while(ch<'0' || ch>'9'){if(ch=='-')w=-1; ch=getchar();}
while(ch>='0' && ch<='9'){s=s*10+ch-'0'; ch=getchar();}
return w*s;
}
void adde(int x,int y)
{
cnt++;
e[cnt].to=y;
e[cnt].next=head[x];
head[x]=cnt;
}
void dfs1(int u,int fa)
{
f[u]=fa; deep[u]=deep[fa]+1; size[u]=1;
for(int i=head[u];i;i=e[i].next)
{
int v=e[i].to;
if(v==fa) continue;
dfs1(v,u);
size[u]+=size[v];
if(size[son[u]]<size[v]) son[u]=v;
}
}
void dfs2(int u,int fa,int tp)
{
top[u]=tp; in[u]=out[u]=++tim;
if(son[u]) dfs2(son[u],u,tp);
for(int i=head[u];i;i=e[i].next)
{
int v=e[i].to;
if(v==fa || v==son[u]) continue;
dfs2(v,u,v);
}
out[u]=tim;
}
void build(int b,int l,int r)
{
tree[b].l=l; tree[b].r=r;
if(l==r)
{
tree[b].sum=0; tree[b].lazy=0; tree[b].add=0;
return ;
}
int mid=(l+r)>>1;
build(b<<1,l,mid);
build(b<<1|1,mid+1,r);
tree[b].sum=tree[b<<1].sum+tree[b<<1|1].sum;
}
void pushdown(int b,int l,int r)
{
if(tree[b].lazy)
{
int mid=(l+r)>>1;
tree[b<<1].add=tree[b<<1|1].add=tree[b].add;
tree[b<<1].sum=(mid-l+1)*tree[b].add; tree[b<<1].lazy=1;
tree[b<<1|1].sum=(r-mid)*tree[b].add; tree[b<<1|1].lazy=1;
tree[b].lazy=0;
}
}
void update(int b,int x,int y,int c)
{
int l=tree[b].l,r=tree[b].r,mid=(l+r)>>1;
if(x<=l && y>=r)
{
tree[b].sum=(r-l+1)*c;
tree[b].lazy=1;
tree[b].add=c;
return ;
}
pushdown(b,l,r);
if(y<=mid) update(b<<1,x,y,c);
else if(x>mid) update(b<<1|1,x,y,c);
else {update(b<<1,x,mid,c); update(b<<1|1,mid+1,y,c);}
tree[b].sum=tree[b<<1].sum+tree[b<<1|1].sum;
}
int query(int b,int x,int y)
{
int l=tree[b].l,r=tree[b].r,mid=(l+r)>>1;
if(x<=l && y>=r) return tree[b].sum;
pushdown(b,l,r);
if(y<=mid) return query(b<<1,x,y);
else if(x>mid) return query(b<<1|1,x,y);
else return query(b<<1,x,mid)+query(b<<1|1,mid+1,y);
}
int solve(int x,int y)
{
int ans=0;
while(top[x]!=top[y])
{
ans+=(in[x]-in[top[x]]+1)-query(1,in[top[x]],in[x]);
update(1,in[top[x]],in[x],1);
x=f[top[x]];
}
ans+=in[x]-in[y]+1-query(1,in[y],in[x]);
update(1,in[y],in[x],1);
return ans;
}
int main()
{
n=read();
for(int i=1;i<n;i++)
{
int x=read();
adde(x+1,i+1);
}
dfs1(1,0);
dfs2(1,0,1);
build(1,1,n);
int q=read();
while(q--)
{
char op[10];
scanf("%s",op);
int x=read();
x++;
if(op[0]=='i') printf("%d\n",solve(x,1));
else
{
printf("%d\n",query(1,in[x],out[x]));
update(1,in[x],out[x],0);
}
}
return 0;
}
洛谷数据不开O2用时5.01s
开O2用时2.44s
吸氧真好用