CF375D Tree and Queries（dsu on tree，小技巧）

题目链接

题意

思路

      明显是 dsu on tree，主要看一下怎么维护答案；

      问出现次数大于 k 的颜色种类数，可开一个数组记录每种颜色的出现次数，再用树状数组或线段树维护，对于每一个次数，有多小种颜色即可，复杂度；

     或者，开一个数组 num[i] 表示出现次数大于等于 i 的颜色种类数，若当前一种颜色 x， 出现次数为 cnt[x]；

     再加一个颜色 x 时，在 cnt[x] 变化之前，num[ cnt[x] + 1 ] + 1，且 num[ cnt[x] ] 不变；

     减一个颜色 x 时，num[ cnt[x] ] - 1，且 num[ cnt[x] - 1 ] 不变；即可；

代码

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

struct node{
	int to,nex;
}edge[200010];
int head[100010],cnt;
int siz[100010],dfn[100010],ndfn[100010],son[100010];
int skp,col[100010],cntcol[100010];
int n,num[100010];
vector<pair<int,int> > q[100010];
int ans[100010];

void addedge(int x,int y){
	edge[++cnt].to=y;
	edge[cnt].nex=head[x];
	head[x]=cnt;
}

void dfs(int p,int f){
	siz[p]=1;
	dfn[p]=++cnt;
	ndfn[cnt]=p;
	for(int i=head[p];i!=-1;i=edge[i].nex){
		int y=edge[i].to;
		if(y==f) continue;
		dfs(y,p);
		if(siz[son[p]]<siz[y]) son[p]=y;
		siz[p]+=siz[y];
	}
}

void get_data(int p,int k){
	for(int i=dfn[p];i<=dfn[p]+siz[p]-1;i++){
		if(i==dfn[skp]){
			i+=siz[skp]-1;
			continue;
		}
		int x=ndfn[i];
		if(k==1){
			cntcol[col[x]]++;
			num[cntcol[col[x]]]++;
		}
		else{
			num[cntcol[col[x]]]--;
			cntcol[col[x]]--;
		}
	}
}

void dsu(int p,int f,bool del){
	for(int i=head[p];i!=-1;i=edge[i].nex){
		int y=edge[i].to;
		if(y==f||y==son[p]) continue;
		dsu(y,p,true);
	}
	if(son[p]){
		dsu(son[p],p,false);
		skp=son[p];
	}
	
	get_data(p,1);
	for(int i=0;i<q[p].size();i++) ans[q[p][i].second]=num[q[p][i].first];
	if(del){
		skp=0;
		get_data(p,-1);
	}
}

int main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	memset(head,-1,sizeof(head));
	int x,y,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>col[i];
	for(int i=1;i<n;i++){
		cin>>x>>y;
		addedge(x,y); addedge(y,x);
	}
	for(int i=1;i<=m;i++){
		cin>>x>>y;
		q[x].push_back({y,i});
	}
	cnt=0;
	dfs(1,0);
	dsu(1,0,0);
	for(int i=1;i<=m;i++) cout<<ans[i]<<"\n";
	return 0;
}