当前位置: 首页 > news >正文

洛谷U639786 树的颜色询问 题解 树上启发式合并(dsu on tree)

题目链接:https://www.luogu.com.cn/problem/U639786

题目大意:

树上每个节点有一个颜色,节点 \(i\) 的颜色是 \(c_i\)

对于每个节点,求它所在的子树中出现次数最多的颜色出现了多少次。

解题思路:

树上启发式合并.模板题。

这里我采用了一种 自己的写法。

一些变量的含义:

  • t[x]:表示颜色 \(x\) 目前出现了多少次;
  • cnt[x]:表示有多少种颜色出现了 \(x\) 次;
  • mx:出现次数最多的颜色出现了多少次;
  • ans[u]:由于是离线处理,所以记录 \(u\) 对应的答案,即 \(u\) 所在的子树中出现次数最多的颜色出现了多少次。

示例程序:

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;int n, c[maxn], t[maxn], cnt[maxn], mx, ans[maxn], sz[maxn], son[maxn];
vector<int> g[maxn];// 新增一个颜色为x的节点
void add_c(int x) {cnt[ t[x] ]--;t[x]++;cnt[ t[x] ]++;mx = max(mx, t[x]);
}// 删除一个颜色为x的节点
void del_c(int x) {cnt[ t[x] ]--;t[x]--;cnt[ t[x] ]++;if (!cnt[mx]) mx--;
}void getsz(int u, int p) {sz[u] = 1;for (auto v : g[u]) {if (v != p) {getsz(v, u);sz[u] += sz[v];if (sz[v] > sz[ son[u] ])son[u] = v;}}
}void dfs_clean(int u, int p) {del_c(c[u]);for (auto v : g[u])if (v != p)dfs_clean(v, u);
}void dfs(int u, int p, bool keep) {for (auto v : g[u])if (v != p && v != son[u])dfs(v, u, false);if (son[u])dfs(son[u], u, true);for (auto v : g[u])if (v != p && v != son[u])dfs(v, u, true);add_c(c[u]);if (!ans[u]) // 第一次访问到的时候是答案ans[u] = mx;if (!keep)dfs_clean(u, p);
}int main() {scanf("%d", &n);for (int i = 1; i <= n; i++)scanf("%d", c+i);for (int i = 1, u, v; i < n; i++) {scanf("%d%d", &u, &v);g[u].push_back(v);g[v].push_back(u);}getsz(1, -1);dfs(1, -1, true);for (int i = 1; i <= n; i++)printf("%d\n", ans[i]);return 0;
}
http://www.jsqmd.com/news/69194/

相关文章:

  • 2025年热门的牛羊肉贴体膜/贴体膜厂家最新实力排行
  • 2025 年雅思培训口碑机构 TOP5 推荐
  • 灵光网页版AI助手,特斯拉集成Grok语音导航,阿里Qwen3-TTS横空出世
  • 软件工程学习日志2025.12.9
  • 2025年口碑好的网架工程/徐州煤棚网架厂家选购指南与推荐
  • 2025雅思培训机构怎么选?这篇攻略帮你避坑+精准提分!
  • 2025年热门的格栅机耙齿用户口碑最好的厂家榜
  • 英语自学工具进化论:告别哑巴英语,走向真实对话时代
  • 2025年比较好的融雪伴热带/高温伴热带厂家实力及用户口碑排行榜
  • 学英语,最好的软件其实是“组合拳”哪些英语软件最有效?
  • 【亲测】AI学术搜索哪家强?试了4款国产顶流,结果完全出乎意料!
  • 详细介绍:[Column] Perplexity 如何构建 AI 版 Google | 模型无关架构 | Vespa AI检索
  • 2025年英语自学软件精选:免费高效,轻松开启学习之旅
  • 2025年全国太阳能路灯厂家五大最新推荐:涵盖太阳能路灯、景观灯、庭院灯、高杆路灯、LED路灯厂家选择指南
  • 2025实测|5款英语学习软件封神!从零基础到流利说全靠它
  • 2025中国薪酬管理系统测评与选型指南:数字化转型下的企业决策
  • Oracle中行转列、列转行
  • 告别英语焦虑!这些英语自学神器让你快速逆袭
  • 2025年比较好的注塑机边粉碎机/粉碎机厂家推荐及选择参考
  • 解锁成人英语学习新方法,这些APP超神啦!
  • Java虚拟机(JVM)面试题(51道含答案) - 实践
  • 102302116_田自豪_作业4
  • Day58(28)-F:\vs_ai_work\vue-tlias-management
  • 解锁全球社交新姿势:7 款免费跨国聊天软件深度测评
  • 随机名字生成器
  • C++进阶:(九)深度剖析unordered_map 与 unordered_set容器 - 指南
  • 2025年口碑好的电解电镀高压直流继电器厂家推荐及选购参考榜
  • SSAS-基本概念 - 指南
  • 跨境电商人的英语逆袭神器,你 get 了吗?
  • 跨境电商英语学习app推荐