CF600E(dsu on tree)

题意:

给你一棵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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值