【CF600E】 Lomsat gelral

CF600E Lomsat gelral

---

Solution

考虑一下子树的问题,我们可以把一棵树的dfn序搞出来,那么子树就是序列上的一段连续的区间.

然后就可以莫队飞速求解了.

但是这题还有\(\Theta(nlog_n)\)的做法.能有\(\Theta(n\sqrt{n})\)的做法要什么\(logn\)的

考虑\(dsu\ on\ tree\),与莫队没有任何区别.

如果不会的话,请自行跳转小Z的袜子并且切掉.

代码实现

#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
using namespace std;
const int N=100010;
int n,front[N],nxt[N<<1],to[N<<1],cnt,siz[N],son[N],num[N],c[N],Max,b[N];
long long ans[N];
pair<int,long long>f[N];
void Add(int u,int v){to[++cnt]=v;nxt[cnt]=front[u];front[u]=cnt;}
void dfs(int u,int fa){
	siz[u]=1;
	for(int i=front[u];i;i=nxt[i]){
		int v=to[i];
		if(v==fa)continue;
		dfs(v,u);siz[u]+=siz[v];
		if(siz[v]>siz[son[u]])son[u]=v;
	}
}
void add(int u,int fa,int opt){
	int &p=num[c[u]];
	f[p].first--;f[p].second-=c[u];
	p+=opt;
	f[p].first++;f[p].second+=c[u];
	if(opt==1)Max=max(Max,p);
	else if(!f[Max].first)Max--;
	for(int i=front[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v!=fa && !b[v])add(v,u,opt);
	}
}
void dfs(int u,int fa,int opt){
	for(int i=front[u];i;i=nxt[i]){
		int v=to[i];
		if(v!=fa && v!=son[u])dfs(v,u,0);
	}
	if(son[u])dfs(son[u],u,1);b[son[u]]=1;
	add(u,fa,1);
	ans[u]=f[Max].second;
	b[son[u]]=0;
	if(!opt)add(u,fa,-1);
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d",&c[i]);
	for(int i=1;i<n;i++){
		int u,v;scanf("%d%d",&u,&v);
		Add(u,v);Add(v,u);
	}
	dfs(1,1);dfs(1,1,1);
	for(int i=1;i<=n;i++)printf("%lld%c",ans[i],i==n?'\n':' ');
	return 0;
}