[NOI2015]软件包管理器————树链剖分

题解:本题主要考查树链剖分,写了好久!
简要题意:树上两个操作,1:查询从该节点到根有多少个结点权值为0,并且把0−x上的所有节点都赋为1。2:查询该节点为根的子树总权值和,并且更新子树全部置为0
1.树链剖分:其实本题主要难点是理解题意,怎么把题目的信息转化为上面的题意。就会发现其实是树链剖分的模板改动一点,区间覆盖。
这里用了输出的优化,总和就为:ans=abs(操作前的总和-操作后的总和)。(原来我没有用,看了题解就改过来了)
代码如下:

#include<iostream>
#include<algorithm>
using namespace std;
struct N
{
	int ans,lazy;
}t[800005];
struct E
{
	int start,to;
}e[800005];
int siz[200005],d[200005],son[200005],h[200005];
int top[200005],rk[200005],f[200005],id[200005];
int n,m,P,num,sum;
void pushup(int p)
{
	t[p].ans=t[p*2].ans+t[p*2+1].ans; 
}
void build(int p,int l,int r)
{
	t[p].lazy=-1;t[p].ans=0;
	if(l==r)return ;
	int mid=l+r>>1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	pushup(p);
	return ;
}
void lai(int p,int l,int r)
{
	int mid=l+r>>1;
	if(t[p].lazy!=-1)
	{
		t[p*2].lazy=t[p*2+1].lazy=t[p].lazy;
		t[p*2].ans=t[p].lazy*(mid-l+1);
		t[p*2+1].ans=t[p].lazy*(r-mid);
		t[p].lazy=-1;
	}
	return ;
}
void change(int p,int l,int r,int x,int y,int z)
{
	if(x>r||y<l)return ;
	if(x<=l&&y>=r)
	{
		t[p].ans=z*(r-l+1);
		t[p].lazy=z;
		return ;
	}
	lai(p,l,r);
	int mid=l+r>>1;
	if(x<=mid)change(p*2,l,mid,x,y,z);
	if(y>mid)change(p*2+1,mid+1,r,x,y,z);
	pushup(p);
	return ;
}
void add(int start,int to)
{
	e[++P].to=to;
	e[P].start=h[start];
	h[start]=P;
}
void dfs1(int p,int fa,int deep)
{
	d[p]=deep;f[p]=fa;siz[p]=1;
	for(int i=h[p];i;i=e[i].start)
	{
		int k=e[i].to;
		if(k==fa)continue;
		dfs1(k,p,deep+1);
		siz[p]+=siz[k];
		if(!son[p]||siz[k]>siz[son[p]])son[p]=k;
	}
}
void dfs2(int p,int tp)
{
	id[p]=++num;rk[num]=p;top[p]=tp;
	if(!son[p])return ;
	dfs2(son[p],tp);
	for(int i=h[p];i;i=e[i].start)
	{
		int k=e[i].to;
		if(k!=son[p]&&k!=f[p])
		dfs2(k,k);
	}
}
void update(int x,int y,int v)
{
	while(top[x]!=top[y])
	{
		if(d[top[x]]<d[top[y]])swap(x,y);
		change(1,1,n,id[top[x]],id[x],v);
		x=f[top[x]];
	}
	if(d[x]>d[y])swap(x,y);
	change(1,1,n,id[x],id[y],v);
}
int main()
{
	cin>>n;
	for(int i=2;i<=n;i++)
	{
		int x;
		cin>>x;x++;
		add(x,i);add(i,x);
	}
	dfs1(1,1,1);dfs2(1,1);
	build(1,1,n);
	cin>>m;
	for(int i=1;i<=m;i++)
	{
		char a[129193];int x;
		cin>>a;cin>>x;x++;
		sum=t[1].ans;
		if(a[0]=='i')
		{
			update(1,x,1);
			cout<<t[1].ans-sum<<endl;
		}
		else if(a[0]=='u')
		{
			change(1,1,n,id[x],id[x]+siz[x]-1,0);
			cout<<sum-t[1].ans<<endl;
		}
	}
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值