「NOI2015」软件包管理器

发表于 | 分类于 | | 浏览量： 次

又是一个树剖题QwQ。

传送门

洛谷P2146

BZOJ4196

题解

不难发现，各个软件包之间的依赖关系构成了一棵树。

所以先把树建出来，每一个节点设一个权值，权值为$0$表示还未安装，权值为$1$表示已安装。

当安装软件$x$的时候，答案为$x$到根节点的路径上没有安装的软件的数量，然后把一整条路径上的节点的权值都设成1。

卸载软件$x$的时候，答案为$x$的子树中已经安装的软件的数量，然后把整棵子树中所有节点的权值都设为0。

然后弄个支持区间覆盖查询区间加和的线段树维护下就好了。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=100005;
int n,m,tot,lnk[maxn],son[maxn*2],w[maxn],nxt[maxn*2],idx,siz[maxn],wson[maxn],fa[maxn],top[maxn],seg[maxn],rev[maxn],ans;char cmd[10];
inline int read()
{
	int ret=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-f;ch=getchar();}
	while(ch>='0'&&ch<='9'){ret=ret*10+ch-'0';ch=getchar();}
	return ret*f;
}
struct SegmentTree
{
	struct Node{int Sum,Tag;}Tree[maxn*4];
	void PushUp(int rt){Tree[rt].Sum=Tree[rt*2].Sum+Tree[rt*2+1].Sum;}
	void PushDown(int rt,int LC,int RC)
	{
		if(Tree[rt].Tag==-1) return;
		Tree[rt*2].Sum=Tree[rt].Tag*LC;
		Tree[rt*2+1].Sum=Tree[rt].Tag*RC;
		Tree[rt*2].Tag=Tree[rt].Tag;
		Tree[rt*2+1].Tag=Tree[rt].Tag;
		Tree[rt].Tag=-1;
	}
	void RangeUpdate(int LL,int RR,int delta,int L=1,int R=n,int rt=1)
	{
		if(LL<=L&&R<=RR)
		{
			Tree[rt].Sum=delta*(R-L+1);
			Tree[rt].Tag=delta;
			return;
		}
		int M=(L+R)/2;
		PushDown(rt,M-L+1,R-M);
		if(LL<=M) RangeUpdate(LL,RR,delta,L,M,rt*2);
		if(M<RR) RangeUpdate(LL,RR,delta,M+1,R,rt*2+1);
		PushUp(rt);
	}
	int QueryRange(int LL,int RR,int L=1,int R=n,int rt=1)
	{
		if(LL<=L&&R<=RR) return Tree[rt].Sum;
		int M=(L+R)/2,ret=0;
		PushDown(rt,M-L+1,R-M);
		if(LL<=M) ret+=QueryRange(LL,RR,L,M,rt*2);
		if(M<RR) ret+=QueryRange(LL,RR,M+1,R,rt*2+1);
		return ret;
	}
}T;
inline void add_e(int x,int y){tot++;son[tot]=y;nxt[tot]=lnk[x];lnk[x]=tot;}
void DFS1(int now)
{
	siz[now]=1;
	for(int i=lnk[now];i;i=nxt[i])
	{
		if(son[i]==fa[now]) continue;
		fa[son[i]]=now;
		DFS1(son[i]);
		siz[now]+=siz[son[i]];
	}
	for(int i=lnk[now];i;i=nxt[i])
		if(son[i]!=fa[now]&&siz[son[i]]>siz[wson[now]])
			wson[now]=son[i];
}
void DFS2(int now)
{
	seg[now]=++idx;rev[seg[now]]=now;
	if(now==1) top[now]=1;
	if(wson[now])
	{
		top[wson[now]]=top[now];
		DFS2(wson[now]);
	}
	for(int i=lnk[now];i;i=nxt[i])
	{
		if(son[i]==fa[now]||son[i]==wson[now]) continue;
		top[son[i]]=son[i];
		DFS2(son[i]);
	}
}
int main()
{
	n=read();
	for(int i=2;i<=n;i++)
	{
		int a=read()+1;
		add_e(a,i);add_e(i,a);
	}
	for(int i=1;i<=n*4;i++) T.Tree[i].Tag=-1;
	DFS1(1);DFS2(1);
	m=read();
	while(m--)
	{
		scanf("%s",cmd);
		int x=read()+1;ans=0;
		if(strcmp(cmd,"install")==0)
		{
			while(x>0)
			{
				ans+=seg[x]-seg[top[x]]+1-T.QueryRange(seg[top[x]],seg[x]);
				T.RangeUpdate(seg[top[x]],seg[x],1);
				x=fa[top[x]];
			}
			printf("%d\n",ans);
		}
		else
		{
			ans=T.QueryRange(seg[x],seg[x]+siz[x]-1);
			T.RangeUpdate(seg[x],seg[x]+siz[x]-1,0);
			printf("%d\n",ans);
		}
	}
	return 0;
}