[块状树] BZOJ 3720 Gty的妹子树

最新推荐文章于 2022-09-23 17:02:26 发布

原创最新推荐文章于 2022-09-23 17:02:26 发布 · 556 阅读

0 ·

CC 4.0 BY-SA版权

块状树专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种基于树分块的动态子树查询算法，通过将树状结构划分为多个块来提高查询效率。该算法适用于动态更新节点值并进行子树范围查询的问题，详细阐述了如何构建和维护这些块，以及如何高效地执行查询操作。

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

动态询问子树问题

树分块！

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;

inline char nc()
{
	static char buf[100000],*p1=buf,*p2=buf;
	if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
	return *p1++;
}

inline void read(int &x)
{
	char c=nc(),b=1;
	for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
	for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

const int N=100005;

struct Block{
	int a[210],n;
	void insert(int x){
		a[++n]=x; int p=n;
		while (p>1 && a[p]<a[p-1]) swap(a[p],a[p-1]),p--;
	}
	void modify(int x,int y){
		int p=lower_bound(a+1,a+n+1,x)-a;
		a[p]=y;
		while (p>1 && a[p]<a[p-1]) swap(a[p],a[p-1]),p--;
		while (p<n && a[p]>a[p+1]) swap(a[p],a[p+1]),p++;
	}
	int Query(int x){
		int p=upper_bound(a+1,a+n+1,x)-a;
		return n-p+1;
	}
}Blo[10005];
int cnt;

struct edge{
	int u,v,next;
};

edge G[N<<4];
int head[N],first[N],inum;

inline void add(int *head,int u,int v,int p){
	G[p].u=u; G[p].v=v; G[p].next=head[u]; head[u]=p;
}

int n,B,a[N];

int fat[N],belong[N];

#define V G[p].v
inline void dfs(int u,int fa){
	fat[u]=fa;
	if (u==1 || Blo[belong[fa]].n==B)
		Blo[belong[u]=++cnt].insert(a[u]),add(first,belong[fa],belong[u],++inum);
	else
		Blo[belong[u]=belong[fa]].insert(a[u]);
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa)
			dfs(V,u);
}

int Y;
int ans=0;

inline void block_dfs(int u){
	ans+=Blo[u].Query(Y);
	for (int p=first[u];p;p=G[p].next)
		block_dfs(V);
}

inline void find(int u,int fa){
	if (a[u]>Y) ans++;
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa)
			if (belong[V]==belong[u])
				find(V,u);
			else
				block_dfs(belong[V]);
}

int main()
{
	int Q,order,iu,iv,lastans=0;
	freopen("t.in","r",stdin);
	freopen("t.out","w",stdout);
	read(n); B=sqrt(n)+1e-7;
	for (int i=1;i<n;i++)
		read(iu),read(iv),add(head,iu,iv,++inum),add(head,iv,iu,++inum);
	for (int i=1;i<=n;i++) read(a[i]);
	dfs(1,0);
	read(Q);
	while (Q--)
	{
		read(order);
		read(iu); read(iv); iu^=lastans; iv^=lastans;
		if (order==0)
		{
			Y=iv;
			ans=0;
			find(iu,fat[iu]);
			printf("%d\n",lastans=ans);
		}
		else if (order==1)
		{
			Blo[belong[iu]].modify(a[iu],iv);
			a[iu]=iv;
		}
		else if (order==2)
		{
			a[++n]=iv;
			add(head,iu,n,++inum),add(head,n,iu,++inum);
			fat[n]=iu;
			if (Blo[belong[iu]].n==B)
				Blo[belong[n]=++cnt].insert(a[n]),add(first,belong[iu],belong[n],++inum);
			else
				Blo[belong[n]=belong[iu]].insert(a[n]);
		}
	}
	return 0;
}