[Codeforces375D]Tree and Queries(莫队算法)

Posted void_f

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[Codeforces375D]Tree and Queries(莫队算法)相关的知识,希望对你有一定的参考价值。

题意:给定一棵树,每个节点有颜色,对于每个询问(u,k)询问以u为根节点的子树下有多少种颜色出现次数>=k

因为是子树,跟dfs序有关,转化为一段区间,可以用莫队算法求解

直接用一个数组统计出现次数>=k的颜色

Code

 

#include <cstdio>
#include <algorithm>
#include <cmath>
#define N 100010
using namespace std;

int n,m,A[N],bl[N],Ans[N],dfn[N],sum[N],tot,head[N],sz[N],col[N],tw[N];
struct node{int to,nex;}e[N*2];
struct info{
	int l,r,k,id;
	friend bool operator <(info a,info b){
		return (bl[a.l]==bl[b.l])?a.r<b.r:a.l<b.l;
	}
}q[N];

inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();}
    return x*f;
}

void upd(int x,int d){
	if(d>0) sum[++col[tw[x]]]++;
	else sum[col[tw[x]]--]--;
}

void Link(int u,int v){
	e[++tot].nex=head[u];e[tot].to=v;head[u]=tot;
}

void dfs(int u,int fa){
	dfn[u]=++tot,sz[u]=1,tw[tot]=A[u];
	for(int i=head[u];i;i=e[i].nex)
		if(e[i].to!=fa) dfs(e[i].to,u),sz[u]+=sz[e[i].to];
}

int main(){
	n=read(),m=read();int blo=sqrt(n);
	for(int i=1;i<=n;++i) A[i]=read(),bl[i]=i/blo+1;
	for(int i=1;i<n;++i){
		int u=read(),v=read();
		Link(u,v),Link(v,u);
	}
	tot=0,dfs(1,0);
	for(int i=1;i<=m;++i){
		int u=read(),k=read();
		q[i]=(info){dfn[u],dfn[u]+sz[u]-1,k,i};
	}
	sort(q+1,q+m+1);
	for(int i=1,l=1,r=0;i<=m;++i){
		for(;l<q[i].l;l++) upd(l,-1);
		for(;l>q[i].l;l--) upd(l-1,1);
		for(;r<q[i].r;r++) upd(r+1,1);
		for(;r>q[i].r;r--) upd(r,-1);
		Ans[q[i].id]=sum[q[i].k];
	}
	for(int i=1;i<=m;printf("%d\n",Ans[i++]));
	return 0;
}

 

以上是关于[Codeforces375D]Tree and Queries(莫队算法)的主要内容,如果未能解决你的问题,请参考以下文章

[Codeforces375D]Tree and Queries(莫队算法)

CodeForces 375D Tree and Queries 莫队||DFS序

CF375D Tree and Queries

题解 CF375D Tree and Queries

题解 CF375D Tree and Queries

codeforces 416B. Appleman and Tree 树形dp