Codeforces 600E dsu on tree 启发式合并

题目:传送门

题意:

给一颗树,计算所有结点的子树,颜色最多的结点的 颜色种类的sum和,子树的定义是当前结点和其所有的子节点。

题解:

我们可以想到一种暴力的解法,就是O(n)遍历所有的结点,然后dfs搜索它的所有子节点,查看颜色出现最多的结点,用一个ans数组维护答案,最后输出ans。显然,这样做的复杂度为O(n^2^)
我们这里用到dsu on tree算法,关于这个算法的详解,可以参考:http://codeforces.com/blog/entry/44351

AC 代码:

#include <bits/stdc++.h>
#define debug(x) cout<<#x<<" = "<<x<<endl;
#define INF 0x3f3f3f3f
using namespace std;

const int maxn = 100010;
vector<int> g[maxn];
int Max,col[maxn];
long long sum,ans[maxn];
/****************dsu on tree 算法************************/
int sz[maxn];
void getsz(int v, int p) {//计算每个结点有多少个子节点
    sz[v] = 1;  
    for (auto u : g[v])
        if (u != p) {
            getsz(u, v);
            sz[v] += sz[u];
        }
}
int cnt[maxn];
bool big[maxn];
void add(int v, int p, int x) {
    cnt[ col[v] ] += x;
    if (cnt[col[v]] > Max)
        sum = col[v], Max = cnt[col[v]];
    else if (cnt[col[v]] == Max)
        sum += col[v];
    for (auto u : g[v])
        if (u != p && !big[u])
            add(u, v, x);
}
void dfs(int v, int p, int keep) {//启发式搜索
    int mx = -1, bigChild = -1;
    for (auto u : g[v])
        if (u != p && sz[u] > mx)
            mx = sz[u], bigChild = u;
    for (auto u : g[v])
        if (u != p && u != bigChild)
            dfs(u, v, 0);  
    if (bigChild != -1)
        dfs(bigChild, v, 1), big[bigChild] = 1;  
    add(v, p, 1);
    if (bigChild != -1)
        big[bigChild] = 0;
    ans[v] = sum;
    if (keep == 0)
        add(v, p, -1), Max = sum = 0;
}
/****************dsu on tree 算法************************/
int main(void) {
    int n, t1, t2;
    cin >> n;
    for (int i = 1; i <=n; i++)
        cin >> col[i];
    for (int i = 1; i < n; i++)
        cin >> t1 >> t2, g[t1].push_back(t2), g[t2].push_back(t1);
    getsz(1, 0);
    dfs(1, 0, 1);
    for (int i = 1; i <= n; i++)
        cout << ans[i] << " ";
    return 0;
}
Python网络爬虫与推荐算法新闻推荐平台:网络爬虫:通过Python实现新浪新闻的爬取,可爬取新闻页面上的标题、文本、图片、视频链接(保留排版) 推荐算法:权重衰减+标签推荐+区域推荐+热点推荐.zip项目工程资源经过严格测试可直接运行成功且功能正常的情况才上传,可轻松复刻,拿到资料包后可轻松复现出一样的项目,本人系统开发经验充足(全领域),有任何使用问题欢迎随时与我联系,我会及时为您解惑,提供帮助。 【资源内容】:包含完整源码+工程文件+说明(如有)等。答辩评审平均分达到96分,放心下载使用!可轻松复现,设计报告也可借鉴此项目,该资源内项目代码都经过测试运行成功,功能ok的情况下才上传的。 【提供帮助】:有任何使用问题欢迎随时与我联系,我会及时解答解惑,提供帮助 【附带帮助】:若还需要相关开发工具、学习资料等,我会提供帮助,提供资料,鼓励学习进步 【项目价值】:可用在相关项目设计中,皆可应用在项目、毕业设计、课程设计、期末/期中/大作业、工程实训、大创等学科竞赛比赛、初期项目立项、学习/练手等方面,可借鉴此优质项目实现复刻,设计报告也可借鉴此项目,也可基于此项目来扩展开发出更多功能 下载后请首先打开README文件(如有),项目工程可直接复现复刻,如果基础还行,也可在此程序基础上进行修改,以实现其它功能。供开源学习/技术交流/学习参考,勿用于商业用途。质量优质,放心下载使用。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值