当前位置: 首页 > 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.gsyq.cn/news/79755.html

相关文章:

  • Webpack与Vite的常用设置及主要差异分析
  • 2025 年雅思培训口碑机构 TOP5 推荐
  • 2025年质量好的平版胶印油墨/胶印油墨厂家最新热销排行
  • 【亲测】AI学术搜索哪家强?试了4款国产顶流,结果完全出乎意料!
  • 102302116_田自豪_作业4
  • 随机名字生成器
  • 跨境电商英语学习app推荐
  • 跨境电商人的英语逆袭神器,你 get 了吗?
  • 251209一天的任务量还是挺多的
  • Gradio界面进行渐变美学设计的提示词 - yi
  • 芯片腾飞
  • 第四次博客作业
  • 2025视觉检测设备权威测评榜单正式发布
  • 【C语言】结构体
  • 什么是 Spring AOP - Higurashi
  • 2025最新油田助剂厂家推荐榜:实力企业赋能油气开发,全国优质供应商精选
  • Linux 中文本显示字体以颜色突出
  • 《Zephyr RTOS 深度学习指南与生成式AI结合方法探讨》第六章 - 详解
  • 自定义拦截器不生效问题记录
  • 退役选手玩儿原神 (1)
  • 电池的荷电状态(SOC)估计
  • nginx保姆及教学
  • 2025 激光焊接机权威榜单出炉!10 大厂家硬核 PK,国产化技术领跑全球
  • Xcode16
  • 大学生必备 APP 清单,错过血亏!
  • 留学必备APP全攻略:从学习到生活,这几款神器让你轻松适应海外生活
  • 在.NET中实现一库多租户(Single Database Multi-Tenancy)模式,主要通过共享数据库但隔离数据的方式实现。
  • 学习机大揭秘:哪个品牌才是孩子的最佳拍档?
  • 权威解析:十大留学机构深度评测与2025精英选择指南
  • VMware 等企业软件固件下载