bzoj4196: [Noi2015]软件包管理器

最新推荐文章于 2018-12-25 09:42:51 发布

*ACoder*

最新推荐文章于 2018-12-25 09:42:51 发布

阅读量423

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/FSAHFGSADHSAKNDAS/article/details/53129764

本文介绍了一种使用树链剖分结合线段树的数据结构解决特定问题的方法。通过该方法，可以有效地处理涉及树形结构中子树更新及查询的操作。文章详细展示了算法实现过程，并附上了完整的C++代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

　　http://www.lydsy.com/JudgeOnline/problem.php?id=4196

题解

　　半节课就写好了。。。结果超时又调了一节课，其实是0的问题，我数组边界设成了零，而节点编号也是从0开始的。以后凡是从0开始编号的题再也不用0当边界了。。

　　树链剖分+线段树，卸载就是把当前子树置零，安装就是把当前点到0节点全设成1。

代码

//树链剖分+线段树
#include <cstdio>
#include <algorithm>
#include <iostream>
#define maxn 100010
using namespace std;
struct segtree
{
	int l, r, sum, set;
	segtree *lch, *rch;
	segtree(){sum=0;set=-1;lch=rch=0;}
}*root;
int N, head[maxn], next[maxn], to[maxn], fa[maxn], tid[maxn], top[maxn], son[maxn],
	size[maxn], tot, tim, l[maxn], r[maxn], deep[maxn];
void adde(int a, int b){to[++tot]=b;next[tot]=head[a];head[a]=tot;}
void input()
{
	int i;
	scanf("%d",&N);
	fa[0]=-1;
	for(i=1;i<N;i++)
	{
		scanf("%d",fa+i);
		adde(fa[i],i);
	}
}
void pushdown(segtree *p)
{
	if(p->set!=-1)
	{
		p->sum=p->set*(p->r-p->l+1);
		if(p->lch)p->lch->set=p->rch->set=p->set;
		p->set=-1;
	}
}
void update(segtree *p)
{
	if(p->lch==0)return;
	pushdown(p->lch),pushdown(p->rch);
	p->sum=p->lch->sum+p->rch->sum;
}
void build(segtree *p, int l, int r)
{
	int mid=(l+r)>>1;
	p->l=l,p->r=r;
	if(l==r)return;
	build(p->lch=new segtree,l,mid);
	build(p->rch=new segtree,mid+1,r);
}
void segset(segtree *p, int l, int r, int w)
{
	pushdown(p);
	int mid=(p->l+p->r)>>1;
	if(l<=p->l and r>=p->r){p->set=w;return;}
	if(l<=mid)segset(p->lch,l,r,w);
	if(r>mid)segset(p->rch,l,r,w);
	update(p);
}
int segsum(segtree *p, int l, int r)
{
	pushdown(p);
	int mid=(l+r)>>1, ans=0;
	if(l<=p->l and r>=p->r){return p->sum;}
	if(l<=mid)ans+=segsum(p->lch,l,mid);
	if(r>mid)ans+=segsum(p->rch,mid+1,r);
	return ans;
}
void dfs1(int pos)
{
	int p;
	size[pos]=1;
	for(p=head[pos];p;p=next[p])
	{
		if(to[p]==fa[pos])continue;
		deep[to[p]]=deep[pos]+1;
		dfs1(to[p]);
		if(son[pos]==0 or size[to[p]]>size[son[pos]])son[pos]=to[p];
		size[pos]+=size[to[p]];
	}
}
void dfs2(int pos, int tp)
{
	int p;
	top[pos]=tp;
	tid[pos]=++tim;
	l[pos]=tim;
	if(son[pos])dfs2(son[pos],tp);
	for(p=head[pos];p;p=next[p])
	{
		if(to[p]==fa[pos] or to[p]==son[pos])continue;
		dfs2(to[p],to[p]);
	}
	r[pos]=tim;
}
void install(int x)
{
	while(x!=-1)segset(root,tid[top[x]],tid[x],1),x=fa[top[x]];
}
int main()
{
	int i, x, q, last=0;
	char type[100];
	input();
	dfs1(0);
	dfs2(0,0);
	build(root=new segtree,1,tim);
	scanf("%d",&q);
	for(i=1;i<=q;i++)
	{
		scanf("%s%d",type,&x);
		if(*type=='i')install(x);
		else segset(root,l[x],r[x],0);
		x=segsum(root,1,tim);
		printf("%d\n",abs(x-last));
		last=x;
	}
	return 0;
}