树上启发式合并典题 U41492 树上数颜色 的数据未免也太弱了吧。。。

void dfs(int u, int father) {
	f[u].insert(c[u]);
	for (int i = h[u]; i; i = ne[i]) {
		int v = e[i];
		if (v == father) continue;
		dfs(v, u);
		for (auto i : f[v])
			f[u].insert(i);
		f[v].clear();
	}
	ans[u] = f[u].size();
}

$O(n ^ 2 \log n)$ 的都能过去。

有没有人提供一下加强版啊？