CF600E——轻重链剖分

传送门
对于每一个点x，可以分别进行O（n）的搜索，但这会超时。可以采用树链剖分（轻重链剖分）进行优化，对于每个x，长久记录它的重儿子，每次只搜索轻儿子，就可以将算法的复杂度优化到 O(nlogn)。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll read(){
	ll a=0,b=getchar(),c=1;
	while(!isdigit(b))c=b=='-'?-1:1,b=getchar();
	while(isdigit(b))a=a*10+b-'0',b=getchar();
	return a*c;
}
const int N=100005;
ll n,ans[N];
vector<int> g[N];
map<int,ll> a[N],f[N];
void dfs(int x,int y){
	for(int i=0;i<g[x].size();i++){
		int z=g[x][i];
		if(z!=y){
			dfs(z,x);
			if(a[z].size()>a[x].size())
				a[x].swap(a[z]),f[x].swap(f[z]);
			for(map<int,ll>::iterator j=a[z].begin();j!=a[z].end();j++){
				int u=(*j).first,v=a[x][u];
				a[x][u]+=(*j).second;
				if(a[x].count(u))f[x][v]-=(*j).first;
				f[x][v+(*j).second]+=(*j).first;
			}
		}
	}
	map<int,ll>::iterator it=f[x].end();
	it--,ans[x]=it->second;
}
int main(){
	n=read();
	for(int i=1;i<=n;i++)
		f[i][1]=read(),a[i][f[i][1]]=1;
	for(int i=2;i<=n;i++){
		int x=read(),y=read();
		g[x].push_back(y),g[y].push_back(x);
	}
	dfs(1,0);
	for(int i=1;i<=n;i++)
		printf("%lld ",ans[i]);
	return 0;
}