[Bzoj4196][Noi2015]软件包管理器---树链剖分+线段树

NOI2015 软件包管理器

 题目描述 

    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。

 输入描述

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

随后一行包含n−1个整数,相邻整数之间用单个空格隔开,分别表示1,2,3,…,n−2,n−1号软件包依赖的软件包的编号。
接下来一行包含1个正整数q,表示询问的总数。
之后q行,每行1个询问。询问分为两种:
installx:表示安装软件包x
uninstallx:表示卸载软件包x
你需要维护每个软件包的安装状态,一开始所有的软件包都处于未安装状态。对于每个操作,你需要输出这步操作会改变多少个软件包的安装状态,随后应用这个操作(即改变你维护的安装状态)。

 输出描述

输出文件包括q行。

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

 

样例输入

7

0 0 0 1 1 5

5

install 5

install 6  

uninstall 1

install 4

uninstall 0 

样例输出

3

1

3

2

2

 

 明显这是一颗树,所以咱就可以上熟练泼粪树链剖分+线段树~

 只需要记录和更新区间内安装了多少个软件就行啦

 更新子树的时候,因为点在线段树上的分布其实是重链优先的DFS序

 所以子树的更新区间就是[ id[x] , id[x]+size[x]-1 ]啦

 交了好几次才A

 前几次是RE

 1e5改成2e5就变成了TLE(然而不知道为什么RE)

 然后发现题目的查询和更改是绑在一起的

 就把查询和修改改到一个函数里

 然后就A啦~

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#define M 200010
using namespace std;
int n,q,size[M],dep[M],fa[M],rk[M],id[M];
int k,cnt,head[M],son[M],top[M],sum[M*2],lazy[M];
struct node {
    int to,next;
}e[M];
void add(int x,int y) {
    e[k].to = y; e[k].next = head[x];
    head[x] = k++;
}
void init() {
    k=0; cnt=0;
    for (int i=0;i<=n;i++) {
        size[i] = son[i] = 0;
        head[i] = lazy[i] = -1;
        sum[i] = sum[i+n] = 0;
    }
}
void dfs1(int s,int d) {
    dep[s] = d;
    size[s] = 1;
    for (int i=head[s];~i;i=e[i].next)
    if ( e[i].to != fa[s] ) {
        dfs1(e[i].to,d+1);
        size[s] += size[e[i].to];
        if ( size[e[i].to] > size[son[s]] )
            son[s] = e[i].to;
    }
}
void dfs2(int s,int t) {
    top[s] = t;
    id[s] = ++cnt;
    rk[cnt] = s;
    if ( !son[s] ) return;
    dfs2(son[s],t);
    for (int i=head[s];~i;i=e[i].next)
    if ( e[i].to != fa[s] && e[i].to != son[s] )
        dfs2(e[i].to,e[i].to);
}
void pushup(int x) {
    sum[x] = sum[x*2] + sum[x*2+1];
}
void pushdown(int l,int r,int x) {
    if ( lazy[x] != -1 ) {
        int mid = (l+r)/2;
        sum[x*2] = (mid-l+1)*lazy[x];
        lazy[x*2] = lazy[x];
        sum[x*2+1] = (r-mid)*lazy[x];
        lazy[x*2+1] = lazy[x];
        lazy[x] = -1;
    }
}
void update(int L,int R,int x,int l,int r,int a) {
    if ( l<=L&&R<=r ) {
        lazy[x] = a;
        sum[x] = a*(R-L+1);
        return;
    }
    pushdown(L,R,x);
    int mid=(L+R)/2;
    if ( l<=mid )
        update(L,mid,x*2,l,r,a);
    if ( mid < r )
        update(mid+1,R,x*2+1,l,r,a);
    pushup(x);
}
void updates(int x) {
    while ( top[x] != 1 ) {
        update(1,n,1,id[top[x]],id[x],1);
        x=fa[top[x]];
    }
    update(1,n,1,1,id[x],1);
}
int query(int L,int R,int x,int l,int r,int a) {
    int ret = 0;
    if ( l<=L&&R<=r ) {
        ret = sum[x];
        sum[x] = a*(R-L+1);
        lazy[x] = a;
        return ret;
    }
    pushdown(L,R,x);
    int mid = (L+R)/2;
    if ( l<=mid )
        ret+=query(L,mid,x*2,l,r,a);
    if ( mid < r)
        ret+=query(mid+1,R,x*2+1,l,r,a);
    pushup(x);
    return ret;
}
int Que(int x) {
    int ret=0;
    while ( top[x] != 1 ) {
        ret+=query(1,n,1,id[top[x]],id[x],1);
        x=fa[top[x]];
    }
    ret += query(1,n,1,1,id[x],1);
    return ret;
}
int main() {
    cin>>n;
    init();
    fa[1] = 1;
    for (int i=2;i<=n;i++) {
        scanf("%d",&fa[i]);
        fa[i]++;
        add(fa[i],i);
    }
    dfs1(1,1);
    dfs2(1,1);
    char op[2];
    int a,ans;
    cin>>q;
    for (int i=1;i<=q;i++) {
        scanf("%s %d",op,&a);
        a++;
        if ( op[0] == 'i' ) {
            ans = dep[a] - Que(a);
            printf("%d\n",ans);
        }
        else {
            ans = query(1,n,1,id[a],id[a]+size[a]-1,0);
            printf("%d\n",ans);
        }
    }
    return 0;
}
View Code

 

posted @ 2019-07-22 18:01  Jachin毅轩  阅读(303)  评论(1编辑  收藏  举报