【树链剖分】[BZOJ 4196][NOI 2014]软件包管理器
题目描述:
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。
样例输入
7
0 0 0 1 1 5
5
install 5
install 6
uninstall 1
install 4
uninstall 0
样例输出
3
1
3
2
3
题目分析:
简直就是裸的树链剖分,主要就是细节问题,两个操作真的没有什么好说的。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int MAXN = 100000;
const int MAXM = MAXN;
struct tnode{
int l, r, h, flag;
}Tree[MAXN*4+10];
void push_down(int x){
if(Tree[x].flag == 1){
Tree[x].flag = 0;
Tree[x*2].flag = Tree[x*2+1].flag = 1;
Tree[x*2].h = Tree[x*2].r - Tree[x*2].l + 1;
Tree[x*2+1].h = Tree[x*2+1].r - Tree[x*2+1].l + 1;
}
else if(Tree[x].flag == -1){
Tree[x].flag = 0;
Tree[x*2].flag = Tree[x*2+1].flag = -1;
Tree[x*2].h = Tree[x*2+1].h = 0;
}
}
void push_up(int x){Tree[x].h = Tree[x*2].h + Tree[x*2+1].h;}
struct node{
int v;
node *next;
}Edges[MAXM*2+10], *ecnt=Edges, *adj[MAXN+10];
int id[MAXN+10], sons[MAXN+10], hson[MAXN+10], idcnt, Endid[MAXN+10];
int Top[MAXN+10], Fa[MAXN+10];
void addedge(int u, int v){
++ecnt;
ecnt->v = v;
ecnt->next = adj[u];
adj[u] = ecnt;
}
void dfs1(int u, int fa){
sons[u] = 1;
for(node *p=adj[u];p;p=p->next){
if(p->v == fa) continue;
dfs1(p->v, u);
sons[u] += sons[p->v];
if(sons[hson[u]] < sons[p->v])
hson[u] = p->v;
}
}
void dfs2(int u, int fa, int Tp){
id[u] = ++idcnt;
if(!Tp) Tp = u;
Top[u] = Tp;
if(hson[u]) dfs2(hson[u], u, Tp);
for(node *p=adj[u];p;p=p->next){
if(p->v == fa) continue;
if(p->v == hson[u]) continue;
dfs2(p->v, u, 0);
}
Endid[u] = idcnt;
}
void build(int u, int l, int r){
Tree[u].l = l, Tree[u].r = r;
Tree[u].h = 0, Tree[u].flag = 0;
if(l == r) return ;
int mid = (l + r) >> 1;
build(u*2, l, mid);
build(u*2+1, mid+1, r);
}
int query(int u, int l, int r){
if(l <= Tree[u].l && Tree[u].r <= r)
return Tree[u].h;
else if(Tree[u].r < l || Tree[u].l > r)
return 0;
else{
push_down(u);
return query(u*2, l, r) + query(u*2+1, l, r);
}
return -1;
}
void Insert(int u, int l, int r, int v){
if(Tree[u].r < l || Tree[u].l > r) return ;
else if(Tree[u].l == Tree[u].r){
if(v == 1) Tree[u].h = 1;
else Tree[u].h = 0;
}else if(l <= Tree[u].l && Tree[u].r <= r){
Tree[u].flag = v;
if(v == 1) Tree[u].h = Tree[u].r - Tree[u].l + 1;
else if(v == -1) Tree[u].h = 0;
}else{
push_down(u);
Insert(u*2, l, r, v);
Insert(u*2+1, l, r, v);
push_up(u);
}
}
void _install(int x){
int mid = id[x], u = x;
int ans = 0;
while(mid){
int Len = mid-id[Top[u]]+1;
int Len2 = query(1, id[Top[u]], mid);
ans += Len-Len2;
Insert(1, id[Top[u]], mid, 1);
if(Len2 > 0) break;
u = Fa[Top[u]];
mid = id[u];
}
printf("%d\n", ans);
}
void _uninstall(int x){
int ans = query(1, id[x], Endid[x]);
Insert(1, id[x], Endid[x], -1);
printf("%d\n", ans);
}
int main(){
int n, xname, m;
char s[101];
scanf("%d", &n);
for(int i=1;i<n;i++){
scanf("%d", &Fa[i+1]);
Fa[i+1]++;
addedge(i+1, Fa[i+1]);
addedge(Fa[i+1], i+1);
}
dfs1(1, -1);
dfs2(1, -1, 0);
build(1, 1, idcnt);
scanf("%d", &m);
for(int i=1;i<=m;i++){
scanf("%s%d", s, &xname);
xname++;
if(s[0] == 'i')
_install(xname);
else _uninstall(xname);
}
return 0;
}