[NOI2015] 软件包管理器 (树链剖分+dfs序+线段树)

改了几个小时终于改完了
题目描述
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
吸氧真好用

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值