标签：dsu no int ed U41492 tree son fa siz

blackpink

\(O(n^2)\)显然不过我们应该优化成\(O(nlogn)\)

采用树上启发式合并

仿照树链剖分的思想，对于每一个位置，我们先处理所有的轻儿子，然后处理重儿子，统计当前节点的答案，最后把轻儿子删掉就可以了。

这样全局一个桶就够用了。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n;
int x,y;
int col[1000005];
struct e{
	int to;
	int ne;
}ed[3000005];
int p;
int head[1000005];
void add(int f,int to){
	p++;
	ed[p].to=to;
	ed[p].ne=head[f];
	head[f]=p;
	return ;
}
int siz[2000005];
int ns;
int ans[2000005];
int tem;
int son[1000005];
int exi[1000005];
void dfs(int no,int fa){
	siz[no]++;
	for(int i=head[no];i;i=ed[i].ne){
		if(ed[i].to==fa) continue;
		dfs(ed[i].to,no);
		siz[no]+=siz[ed[i].to];
		if(siz[ed[i].to]>siz[son[no]]){
			son[no]=ed[i].to;
		}
	}
}
void work(int no,int fa,int ke){
	if(!exi[col[no]]){
		tem++;
	}
	exi[col[no]]+=ke;
	for(int i=head[no];i;i=ed[i].ne){
		int v=ed[i].to;
		if(v==fa||v==ns) continue;
		work(v,no,ke);
	}
}
void dsu(int no,int fa,int f){
	for(int i=head[no];i;i=ed[i].ne){
		if(ed[i].to==fa||ed[i].to==son[no]) continue;
		dsu(ed[i].to,no,0);
	}
	if(son[no]) {
		dsu(son[no],no,1);
		ns=son[no];
	}
	work(no,fa,1);
	ns=0;
	ans[no]=tem;
	if(!f){
		work(no,fa,-1);
		tem=0;
	}
	
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;++i){
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	for(int i=1;i<=n;++i){
		scanf("%d",&col[i]);
	}
	dfs(1,0);
	//cout<<2344;
	dsu(1,0,1);
	scanf("%d",&n);
	for(int i=1;i<=n;++i){
		scanf("%d",&x);
		printf("%d\n",ans[x]);
	}
	return 0;
}

标签：dsu,no,int,ed,U41492,tree,son,fa,siz
来源： https://www.cnblogs.com/For-Miku/p/15110431.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。