BZOJ4372烁烁的游戏 动态树分治+线段树

Posted CQzhangyu

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了BZOJ4372烁烁的游戏 动态树分治+线段树相关的知识,希望对你有一定的参考价值。

【BZOJ4372】烁烁的游戏

Description

背景:烁烁很喜欢爬树,这吓坏了树上的皮皮鼠。
题意:
给定一颗n个节点的树,边权均为1,初始树上没有皮皮鼠。
烁烁他每次会跳到一个节点u,把周围与他距离不超过d的节点各吸引出w只皮皮鼠。皮皮鼠会被烁烁吸引,所以会一直待在节点上不动。
烁烁很好奇,在当前时刻,节点u有多少个他的好朋友---皮皮鼠。
大意:
给一颗n个节点的树,边权均为1,初始点权均为0,m次操作:
Q x:询问x的点权。
M x d w:将树上与节点x距离不超过d的节点的点权均加上w。

Input

第一行两个正整数:n,m
接下来的n-1行,每行三个正整数u,v,代表u,v之间有一条边。
接下来的m行,每行给出上述两种操作中的一种。

Output

对于每个Q操作,输出当前x节点的皮皮鼠数量。

Sample Input

7 6
1 2
1 4
1 5
2 3
2 7
5 6
M 1 1 2
Q 5
M 2 2 3
Q 3
M 1 2 1
Q 2

Sample Output

2
3
6

HINT

数据范围:
n,m<=10^5,|w|<=10^4
注意:w不一定为正整数,因为烁烁可能把皮皮鼠吓傻了。

题解:动态点分治+线段树裸题,每个节点开两棵线段树维护它子树中的以及它的父亲要从它中减去的即可。

注意特判边界的问题。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int maxn=100010;
int n,m,tot,ans,cnt,mn,rt;
struct sag
{
	int ls,rs,sum;
}s[maxn*150];
int to[maxn<<1],next[maxn<<1],head[maxn],siz[maxn],dep[maxn],pos[maxn],md[20][maxn<<1],fa[maxn],Log[maxn<<1];
int r1[maxn],r2[maxn];
bool vis[maxn];
char str[5];
inline void add(int a,int b)
{
	to[cnt]=b,next[cnt]=head[a],head[a]=cnt++;
}
void getrt(int x,int fa)
{
	int i,tmp=0;
	siz[x]=1;
	for(i=head[x];i!=-1;i=next[i])	if(!vis[to[i]]&&to[i]!=fa)	getrt(to[i],x),siz[x]+=siz[to[i]],tmp=max(tmp,siz[to[i]]);
	tmp=max(tmp,tot-siz[x]);
	if(tmp<mn)	mn=tmp,rt=x;
}
void solve(int x)
{
	vis[x]=1;
	for(int i=head[x];i!=-1;i=next[i])	if(!vis[to[i]])
		tot=siz[to[i]],mn=1<<30,getrt(to[i],x),fa[rt]=x,solve(rt);
}
void dfs(int x,int fa)
{
	md[0][++pos[0]]=dep[x],pos[x]=pos[0];
	for(int i=head[x];i!=-1;i=next[i])	if(to[i]!=fa)
		dep[to[i]]=dep[x]+1,dfs(to[i],x),md[0][++pos[0]]=dep[x];
}
inline int getmin(int a,int b)
{
	a=pos[a],b=pos[b];
	if(a>b)	swap(a,b);
	int k=Log[b-a+1];
	return min(md[k][a],md[k][b-(1<<k)+1]);
}
inline int dis(int a,int b)
{
	return dep[a]+dep[b]-2*getmin(a,b);
}
void updata(int l,int r,int &x,int a,int b)
{
	if(a<0)	return ;
	if(!x)	x=++tot;
	s[x].sum+=b;
	if(l==r)	return ;
	int mid=(l+r)>>1;
	if(a<=mid)	updata(l,mid,s[x].ls,a,b);
	else	updata(mid+1,r,s[x].rs,a,b);
}
int query(int l,int r,int x,int a,int b)
{
	if(!x||(a<=l&&r<=b))	return s[x].sum;
	int mid=(l+r)>>1;
	if(b<=mid)	return	query(l,mid,s[x].ls,a,b);
	if(a>mid)	return	query(mid+1,r,s[x].rs,a,b);
	return query(l,mid,s[x].ls,a,b)+query(mid+1,r,s[x].rs,a,b);
}
inline int rd()
{
	int ret=0,f=1;	char gc=getchar();
	while(gc<‘0‘||gc>‘9‘)	{if(gc==‘-‘)	f=-f;	gc=getchar();}
	while(gc>=‘0‘&&gc<=‘9‘)	ret=ret*10+(gc^‘0‘),gc=getchar();
	return ret*f;
}
int main()
{
	n=rd(),m=rd();
	int i,j,x,y,a,b,u;
	memset(head,-1,sizeof(head));
	for(i=1;i<n;i++)	a=rd(),b=rd(),add(a,b),add(b,a);
	dfs(1,0),tot=n,mn=1<<30,getrt(1,0),solve(rt),tot=0;
	for(j=1;(1<<j)<=2*n-1;j++)	for(i=1;i+(1<<j)-1<=2*n-1;i++)	md[j][i]=min(md[j-1][i],md[j-1][i+(1<<(j-1))]);
	for(i=2;i<=2*n-1;i++)	Log[i]=Log[i>>1]+1;
	for(i=1;i<=m;i++)
	{
		scanf("%s",str);
		if(str[0]==‘Q‘)
		{
			u=rd(),ans=0;
			for(x=u;x;x=y)
			{
				y=fa[x];
				ans+=query(0,n,r1[x],dis(u,x),n);
				if(y)	ans-=query(0,n,r2[x],dis(u,y),n);
			}
			printf("%d\n",ans);
		}
		else
		{
			u=rd(),a=rd(),b=rd();
			for(x=u;x;x=y)
			{
				y=fa[x];
				updata(0,n,r1[x],min(n,a-dis(u,x)),b);
				if(y)	updata(0,n,r2[x],min(n,a-dis(u,y)),b);
			}
		}
	}
	return 0;
}//7 2 1 2 1 4 1 5 2 3 2 7 5 6 M 1 2 1 Q 2

以上是关于BZOJ4372烁烁的游戏 动态树分治+线段树的主要内容,如果未能解决你的问题,请参考以下文章

bzoj 4372: 烁烁的游戏 动态点分治_树链剖分_线段树

bzoj4372 烁烁的游戏

bzoj4372烁烁的游戏

[BZOJ4372]烁烁的游戏

bzoj 4372 烁烁的游戏 —— 点分治+树状数组

BZOJ 4372 烁烁的游戏