题目链接
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。
Input
输入文件的第1行包含1个正整数n,表示软件包的总数。软件包从0开始编号。
随后一行包含n−1个整数,相邻整数之间用单个空格隔开,分别表示1,2,3,…,n−2,n−1号软件包依赖的软件包的编号。
接下来一行包含1个正整数q,表示询问的总数。
之后q行,每行1个询问。询问分为两种:
installx:表示安装软件包x
uninstallx:表示卸载软件包x
你需要维护每个软件包的安装状态,一开始所有的软件包都处于未安装状态。对于每个操作,你需要输出这步操作会改变多少个软件包的安装状态,随后应用这个操作(即改变你维护的安装状态)。
Output
输出文件包括q行。
输出文件的第i行输出1个整数,为第i步操作中改变安装状态的软件包数。
Sample Input
7
0 0 0 1 1 5
5
install 5
install 6
uninstall 1
install 4
uninstall 0
Sample Output
3
1
3
2
3
HINT
一开始所有的软件包都处于未安装状态。
安装 5 号软件包,需要安装 0,1,5 三个软件包。
之后安装 6 号软件包,只需要安装 6 号软件包。此时安装了 0,1,5,6 四个软件包。
卸载 1 号软件包需要卸载 1,5,6 三个软件包。此时只有 0 号软件包还处于安装状态。
之后安装 4 号软件包,需要安装 1,4 两个软件包。此时 0,1,4 处在安装状态。
最后,卸载 0 号软件包会卸载所有的软件包。
n=100000
q=100000
题目大意
给定一棵初始权值全为0的树,定义一下两种操作。
1.给定u节点,将根到u路径上的所有点赋值为1,并输出修改了多少个点。
2.给定u节点,将以u为根的子树中的所有点赋值为0,并输出修改了多少个节点。
题解
先树剖。
第一个操作用基本的树剖找路径做法就好。考虑到一棵子树的树剖序一定是一段连续的序列,记录下每个节点为根的子树中最大的树剖序,直接操作即可。
线段树上的操作就是区间赋值和查询。
一开始在树剖记录子树最大树剖序时叶子节点的没有记录下来,疯狂RE。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define lc (o<<1)
#define rc (o<<1|1)
using namespace std;
const int N = 100000 + 10;
int to[N], nxt[N], hd[N], tot;
int dep[N], siz[N], fa[N], son[N];
int pos[N], bl[N], ed[N];
int n, m, sz;
struct Seg{
int l, r, cov, sum, siz;
}t[N<<2];
void insert(int u, int v){ to[++tot] = v; nxt[tot] = hd[u]; hd[u] = tot; fa[v] = u; }
inline void in(int &x){
x = 0; int f = 1; char c = getchar();
while(c < '0' || c > '9') { if(c == '-') f = -1; c = getchar(); }
while(c >= '0' && c <= '9') { x = x * 10 + c - '0'; c = getchar(); }
x *= f;
}
void init(){
in(n);
int u;
for(int i = 2; i <= n; i++) in(u), insert(u + 1, i);
}
void dfs1(int x){
siz[x] = 1;
for(int i = hd[x]; i; i = nxt[i]){
dep[to[i]] = dep[x] + 1;
dfs1(to[i]);
siz[x] += siz[to[i]];
if(siz[to[i]] > siz[son[x]]) son[x] = to[i];
}
}
void dfs2(int x, int chain){
int k = son[x];
pos[x] = ++sz;
bl[x] = chain;
if(k == 0) { ed[x] = sz; return; }
dfs2(k, chain);
for(int i = hd[x]; i; i = nxt[i])
if(to[i] != k) dfs2(to[i], to[i]);
ed[x] = sz;
}
void build(int o, int l, int r){
t[o].l = l; t[o].r = r; t[o].cov = -1; t[o].sum = 0;
t[o].siz = 1;
if(l == r) return;
int mid = (l + r) >> 1;
build(lc, l, mid);
build(rc, mid+1, r);
t[o].siz = t[lc].siz + t[rc].siz;
}
void pup(int o){
t[o].sum = t[lc].sum + t[rc].sum;
}
void pdw(int o){
if(t[o].cov != -1){
t[lc].sum = t[lc].siz * t[o].cov;
t[rc].sum = t[rc].siz * t[o].cov;
t[lc].cov = t[rc].cov = t[o].cov;
t[o].cov = -1;
}
}
void change(int o, int x, int y, int z){
int l = t[o].l, r = t[o].r, mid = (l + r) >> 1;
if(l == x && r == y){
t[o].sum = t[o].siz * z;
t[o].cov = z;
return;
}
pdw(o);
if(y <= mid) change(lc, x, y, z);
else if(x > mid) change(rc, x, y, z);
else change(lc, x, mid, z), change(rc, mid+1, y, z);
pup(o);
}
int query(int o, int x, int y){
int l = t[o].l, r = t[o].r, mid = (l + r) >> 1;
if(l == x && r == y) return t[o].sum;
pdw(o);
if(y <= mid) return query(lc, x, y);
else if(x > mid) return query(rc, x, y);
else return query(lc, x, mid) + query(rc, mid+1, y);
}
void solve(int x, int y){
int sum = 0, num = dep[y] + 1;
while(bl[x] != bl[y]){
if(dep[bl[x]] < dep[bl[y]]) swap(x, y);
sum += query(1, pos[bl[x]], pos[x]);
change(1, pos[bl[x]], pos[x], 1);
x = fa[bl[x]];
}
if(pos[x] > pos[y]) swap(x, y);
sum += query(1, pos[x], pos[y]);
change(1, pos[x], pos[y], 1);
printf("%d\n", num - sum);
}
void div(){
dfs1(1);
dfs2(1, 1);
}
void work(){
in(m);
char s[20]; int x;
build(1, 1, n);
for(int i = 1; i <= m; i++){
scanf("%s", s);
in(x); x++;
if(s[0] == 'i') solve(1, x);
else{
printf("%d\n", query(1, pos[x], ed[x]));
change(1, pos[x], ed[x], 0);
}
}
}
int main(){
init();
div();
work();
return 0;
}