CF 375D Tree and Queries(子树询问,转化为序列莫队)

题意:给定一棵树,每个点有颜色,m个询问vi,ki,问以vi为根的子树中有多少种颜色出现的次数>=ki。首先我们可以通过dfn把子树询问变成序列询问。(不会看代码,一颗子树中的顶点的dfn一定是连续的,画画就知道了)然后就是经典的莫队了,至于ans怎么更新,见代码啦~

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 100005
int n,m,block,aa[N],a[N],f[N],h[N],num=0,ANS[N],ans[N];
int in[N],out[N],dfn=0,fa[N];
bool vis[N];
struct query{
	int l,r,k,id,block;
}q[N];
struct edge{
	int to,next;
}data[N];
inline int read(){
	int x=0;char ch=getchar();
	while(ch<'0'||ch>'9') ch=getchar();
	while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
	return x;
}
void dfs(int x){
	in[x]=++dfn;
	for(int i=h[x];i;i=data[i].next){
		int y=data[i].to;
		if(fa[x]==y) continue;
		fa[y]=x;dfs(y);
	}
	out[x]=dfn;
}
inline bool cmp(query x,query y){
	return x.block==y.block?x.r<y.r:x.block<y.block;
}
inline void update(int x){
	//cnt->cnt-1 只会影响ans[cnt];cnt->cnt+1 只会影响ans[cnt+1] 
	if(vis[x]) ans[f[a[x]]--]--;
	else ans[++f[a[x]]]++;
	vis[x]^=1;
}
int main(){
//	freopen("a.in","r",stdin);
	n=read();m=read();block=sqrt(n);
	for(int i=1;i<=n;++i) aa[i]=read();
	for(int i=1;i<n;++i){
		int x=read();int y=read();
		data[++num].to=y;data[num].next=h[x];h[x]=num;
		data[++num].to=x;data[num].next=h[y];h[y]=num;
	}
	dfs(1);
	//把颜色映射到dfn序上去,转化为序列上莫队 
	for(int i=1;i<=n;++i) a[in[i]]=aa[i];
	for(int i=1;i<=m;++i){
		int x=read();q[i].l=in[x];q[i].r=out[x];
		q[i].k=read();q[i].id=i;
		q[i].block=(q[i].l-1)/block;
	}
	sort(q+1,q+m+1,cmp);
	int l=1,r=0;
	for(int i=1;i<=m;++i){
		for(;l<q[i].l;++l) update(l);
		for(;l>q[i].l;--l) update(l-1);
		for(;r<q[i].r;++r) update(r+1);
		for(;r>q[i].r;--r) update(r);
		ANS[q[i].id]=ans[q[i].k];
	}
	for(int i=1;i<=m;++i) printf("%d\n",ANS[i]);
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值