题意:
给你一棵n个点的有根树,以1为根,每个点有一个颜色。对于一颗以v为根的子树,其出现次数最多的颜色,为dominating顶点v这个子树的颜色,可能有多个这样的颜色。求以n个顶点分别为根的子树中,dominating 的颜色和。
思路:
一个很好想的暴力思路就是,dfs n个点,统计该点为根的子树中出现的颜色次数。时间、空间都是O(N^2)的。
我们可以每次统计完一个点后清空cnt数组,将空间降到O(N)。
继续优化,很显然,某个点的ans应该与其所有孩子的ans有关,所以我们应该尽量将其孩子的信息给用上。
仔细想一下上面暴力的过程,我们会发现其实在清除的时候,最后一棵子树是没必要清的,我们可以把它对cntcnt数组的贡献一直传上去。于是我们想到了如果人为地使重儿子是最后一棵子树的话,岂不美哉。于是我们在dfsdfs时最后访问重儿子就行了,然后复杂度就被优化到O(nlogn)了。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
int n,col[maxn];
vector<int> G[maxn];
ll ans[maxn];
int sz[maxn],son[maxn],cnt[maxn];
void dfs1(int u,int f){//get son 重孩子
sz[u]=1;
for(auto v : G[u]){
if(v==f) continue;
dfs1(v,u);
sz[u]+=sz[v];
i