P5311-[Ynoi2011]成都七中【点分树,树状数组】

正题

题目链接:https://www.luogu.com.cn/problem/P5311

---

题目大意

给出(n)个点的一棵树，每个节点有一个颜色，(m)次询问提出区间([l,r])的点构成的生成子图中(x)所在连通块的颜色数。

(1leq n,m,a_ileq 10^5)

---

解题思路

用点分树解决本题是很妙的想法。/bx

考虑点分树如何解决，对于一个询问(l,r,x)，如果(x)在点分树上的一个祖先(y)满足(x)到(y)的路径都是([l,r])的点，那么此时在(y)的点分子树上的所有节点(z)都满足如果(y)到(z)的路径都是([l,r])的节点，那么(x)到(z)的也是。具体原因很好理解，因为两条路径重复的那一段路已经满足条件了。

那么我们此时就可以将一个条件拆分成两个条件挂在(y)上了，具体地我们对于每个询问找到(x)点分树上深度最小的祖先(y)满足(x)到(y)的路径上都是([l,r])的节点，然后把这个询问挂在这个点上。

然后我们暴力枚举所有点然后处理它的点分树子树，假设现在枚举到(x)点，我们把它的所有儿子按照(x)到它们的路径上的最小值从大到小询问，然后挂在(x)点上的询问按照(l)从大到小排序。

然后暴力遍历，记录每个颜色最小的(mx)表示(y)到这个颜色的点需要经过的路径最大值，然后把权值丢到树状数组上查询即可。

时间复杂度：(O(nlog^2 n))

---

code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define lowbit(x) (x&-x)
using namespace std;
const int N=1e5+10;
struct edge{
	int to,next;
}a[N<<1];
struct node{
	int x,fa,l,r;
};
int n,m,tot,num,root,ls[N],c[N];
int siz[N],f[N],t[N],lat[N],ans[N];
vector<node> anc[N],son[N],q[N];
bool v[N];
void addl(int x,int y){
	a[++tot].to=y;
	a[tot].next=ls[x];
	ls[x]=tot;return;
}
void Groot(int x,int fa){
	siz[x]=1;f[x]=0;
	for(int i=ls[x];i;i=a[i].next){
		int y=a[i].to;
		if(y==fa||v[y])continue;
		Groot(y,x);siz[x]+=siz[y];
		f[x]=max(f[x],siz[y]);
	}
	f[x]=max(f[x],num-siz[x]);
	if(f[x]<f[root])root=x;
	return;
}
void dfs(int x,int fa,int fr,int mi=1e9,int mx=-1e9){
	mi=min(x,mi);mx=max(x,mx);
	anc[x].push_back((node){x,fr,mi,mx});
	son[fr].push_back((node){x,fr,mi,mx});
	for(int i=ls[x];i;i=a[i].next){
		int y=a[i].to;
		if(y==fa||v[y])continue;
		dfs(y,x,fr,mi,mx);
	}
	return;
}
void Build(int x){
	int S=num;v[x]=1;dfs(x,0,x);
	for(int i=ls[x];i;i=a[i].next){
		int y=a[i].to;
		if(v[y])continue;
		num=(siz[y]>siz[x])?(S-siz[x]):siz[y];
		root=0;Groot(y,x);Build(root);
	}
	return;
}
bool cmp(node x,node y)
{return x.l>y.l;}
void Change(int x,int val){
	while(x<=n){
		t[x]+=val;
		x+=lowbit(x);
	}
	return;
}
int Ask(int x){
	int ans=0;
	while(x){
		ans+=t[x];
		x-=lowbit(x);
	}
	return ans;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&c[i]);
	for(int i=1;i<n;i++){
		int x,y;
		x=i,y=i+1;
		scanf("%d%d",&x,&y);
		addl(x,y);addl(y,x);
	}
	f[0]=n+1;num=n;Groot(1,1);
	Build(root);
	for(int t=1;t<=m;t++){
		int l,r,x;
		scanf("%d%d%d",&l,&r,&x);
		for(int i=0;i<anc[x].size();i++)
			if(anc[x][i].l>=l&&anc[x][i].r<=r)
				{q[anc[x][i].fa].push_back((node){x,t,l,r});break;}
	}
	for(int i=0;i<N;i++)lat[i]=n+1;
	for(int p=1;p<=n;p++){
		sort(q[p].begin(),q[p].end(),cmp);
		sort(son[p].begin(),son[p].end(),cmp);
		int z=0;
		for(int i=0;i<son[p].size();i++){
			node x=son[p][i];
			while(z<q[p].size()&&q[p][z].l>x.l)
			{node y=q[p][z];ans[y.fa]=Ask(y.r);z++;}
			Change(lat[c[x.x]],-1);
			lat[c[x.x]]=min(lat[c[x.x]],x.r);
			Change(lat[c[x.x]],1);
		}
		while(z<q[p].size())
		{node y=q[p][z];ans[y.fa]=Ask(y.r);z++;}
		for(int i=0;i<son[p].size();i++){
			node x=son[p][i];
			Change(lat[c[x.x]],-1);
			lat[c[x.x]]=n+1;
		}
	}
	for(int i=1;i<=m;i++)
		printf("%d
",ans[i]);
	return 0;
}