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

树的重心

首先 \(n\) 为奇数所以只有一个重心 \(rt\),以它为根,对于 \(x \neq rt\),若 \(x\) 成为割掉某边后的重心,则这条边一定不在 \(x\) 子树内,设割掉后另一子树大小为 \(S\),记 \(g_x = \max_{y \in son_x} sz_y\),则有两个限制:

\[2 \times (n - S - sz_x) \le n - S \]

\[2 \times g_x \le n - S \]

得到:

\[n - 2sz_x \le S \le n - 2g_x \]

且要求边不在 \(x\) 子树内,数据结构维护即可。

\(x = rt\) 可以在 dfs 途中处理。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
// typedef __int128 i128;
typedef pair<int, int> pii;
const int N = 3e5 + 10, mod = 998244353;
template<typename T>
void dbg(const T &t) { cout << t << endl; }
template<typename Type, typename... Types>
void dbg(const Type& arg, const Types&... args) {#ifdef ONLINE_JUDGEreturn ;#endifcout << arg << ' ';dbg(args...);
}   
struct Query {int t, l, r;
};
int T, n, cg, sz[N], g[N], dfn[N], idx, m1, m2, in[N];
ll ans;
vector<int>e[N];
vector<Query>q[N];
struct Fenwick {int tr[N];#define lowbit(x) (x & -x)void clear() { for (int i = 1; i <= n; i++) tr[i] = 0; }void add(int x, int v) {if (!x) return ;while (x <= n) {tr[x] += v;x += lowbit(x);}}int qry(int x) {if (x < 0) return 0;int res = 0;while (x) {res += tr[x];x -= lowbit(x);}return res;}int ask(int l, int r) {if (l > r) return 0;return qry(r) - qry(l - 1);}
} t1, t2;
inline void dfs1(int u, int fa) {sz[u] = 1; g[u] = 0;dfn[u] = ++idx;for (auto v : e[u]) if (v != fa) {dfs1(v, u);sz[u] += sz[v];g[u] = max(g[u], sz[v]);}if (!cg && max(g[u], n - sz[u]) * 2 <= n) cg = u;
}
inline void dfs2(int u, int fa) {t1.add(sz[fa], -1);t1.add(n - sz[u], 1);if (u != cg) {int l = n - sz[u] * 2, r = n - g[u] * 2;ans += (ll)u * (t1.ask(l, r) + t2.ask(l, r));if (in[fa]) in[u] = 1;if (in[u]) ans += cg * (sz[u] <= n - sz[m2] * 2);else ans += cg * (sz[u] <= n - sz[m1] * 2);}t2.add(sz[u], 1);for (auto v : e[u]) if (v != fa) {dfs2(v, u);}t1.add(sz[fa], 1);t1.add(n - sz[u], -1);if (u != cg) {int l = n - sz[u] * 2, r = n - g[u] * 2;ans -= (ll)u * t2.ask(l, r);}
}
// 这里有个猎奇的点,就是 cg 为根的时候本来给 0 加一是会出问题的,但是这里由于 sz[0] = 0 恰好抵消了,当然我的意思是 xht 的写法是对的,因为 BIT 里给 x++ 了,但是不加还是会似 
int main() {// freopen("data.in", "r", stdin);// freopen("data.out", "w", stdout);ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);cin >> T;while (T--) {ans = 0;cin >> n;cg = m1 = m2 = 0;t1.clear(); t2.clear();for (int i = 1; i <= n; i++) e[i].clear(), q[i].clear(), in[i] = 0;for (int i = 1, u, v; i < n; i++) {cin >> u >> v;e[u].push_back(v); e[v].push_back(u);}dfs1(1, 0);idx = 0;dfs1(cg, 0);for (auto v : e[cg]) {if (sz[v] > sz[m1]) m2 = m1, m1 = v;else if (sz[v] > sz[m2]) m2 = v;}in[m1] = 1;for (int i = 1; i <= n; i++) t1.add(sz[i], 1);dfs2(cg, 0);cout << ans << '\n';}return 0;
}
http://www.rkmt.cn/news/116363.html

相关文章:

  • Obsidian Tasks插件终极指南:5步构建高效任务管理系统
  • AzerothCore魔兽世界服务器:3分钟搭建完整开发环境终极指南
  • NGO-LSTM回归预测:北方苍鹰算法优化长短期记忆神经网络的数据预测模型
  • 2025年国内十大抖音小店代运营公司权威推荐,云麦电商位居榜首 - 深度智识库
  • Momo Code Sec Inspector Java 完整使用指南
  • 【第61套】年度最难!Top1出炉!
  • apache echarts数据点重影或 Cannot read properties of undefined (reading type)错误问题
  • Emiya 家今天的饭
  • 3分钟玩转Venera漫画阅读器:全平台安装配置与使用技巧分享
  • 广告定制行业排行与选择指南,电梯电子屏广告/应援广告/地铁站广告/电梯广告/社区广告/候车亭广告/明星应援广告广告采购选哪家 - 品牌推荐师
  • EmotiVoice语音合成历史版本回顾:从v0.1到v2.0的重大升级
  • 2025年全氟聚醚生产企业排行榜,新测评精选全氟聚醚专业制造 - mypinpai
  • ScriptHookV完整指南:5分钟打造个性化GTA V游戏世界
  • 2025年办公室装修团队哪家更值得选?上海五大公装企业技术实 - myqiye
  • Redis最佳实践--键值设计
  • EmotiVoice语音合成抗噪能力优化路径探索
  • 建设项目经济评价的指标(笔记2)
  • 实战分享:如何构建东南亚高并发跑腿配送系统
  • 2025河南GEO优化机构TOP5权威推荐:资质齐全老牌公司 - myqiye
  • 2025湖州婚纱摄影推荐指南:聚焦太湖韵味的高口碑机构榜单 - 提酒换清欢
  • LLaMA-Factory 答疑系列二:高频问题 + 官方解决方案,建议收藏备用
  • ssh命令解析
  • Proxyman 问题
  • 域控操作三:更换域用户桌面背景
  • 如何快速配置OpenWrt访问控制:家庭网络管理完整指南
  • 通用智能+商业落地:原力无限全场景应用选择指南 - 资讯焦点
  • 函数调用
  • 如何3步搞定多模态Agent的复杂依赖?Docker多阶段构建深度解读
  • 2025年直流分流器订制厂家推荐榜单:分流器/车规分流器/储能分流器供应商精选 - 品牌推荐官
  • 智能视频水印清除技术:轻松去除静态水印的完整指南