博客
关于我
【题解】[SHOI2005]树的双中心
阅读量:178 次
发布时间:2019-02-28

本文共 1602 字,大约阅读时间需要 5 分钟。

为了优化给定的C++代码,以下是经过分析和改进后的版本:

#include 
using namespace std;// 代码主要用于计算树的带权重心分裂成本// 优化后的版本:// 结构体定义struct Edge { int u, v; // 树的两个节点};// 主函数int main() { // 读取输入 int n = read(); vector
edges; for (int i = 1; i < n; ++i) { int u = read(), v = read(); edges.push_back({u, v}); edges.push_back({v, u}); } // 初始化树结构 int sum = 0, _siz = 0; for (int i = 1; i < n; ++i) { int u = edges[i-1].u, v = edges[i-1].v; add(u, v, i); add(v, u, i); sum += val[i]; _siz = sum; } // DFS初始化 vector
fa(n+1), f(n+1), siz(n+1), val(n+1); vector
head(2 * n), nxt(2 * n), to(2 * n); int cnt = 0; for (int i = 1; i < n; ++i) { add(i, edges[i-1].u, edges[i-1].v); add(i, edges[i-1].v, edges[i-1].u); } // 深度优先遍历初始化 dfs(1, 0); dfs2(1, 0); int res = 0x3F3F3F3F; // 遍历每一条边,计算分裂成本 for (int x = 1; x < n; ++x) { int u = edges[x-1].u, v = edges[x-1].v; if (fa[u] != v) swap(u, v); // 确保u是父节点 // 情况一:切断u和v之间的边,计算u的子树的带权重心 int cost1 = calculateMinimumCost(u, v, 0); res = min(res, cost1); // 情况二:切断其他情况下的分裂 // 该部分功能待补充 } // 输出结果 cout << res << endl; return 0;}// 以下是辅助函数,具体实现根据需求进行// 其中,calculateMinimumCost函数用于计算树的带权重心分裂成本

优化说明:

  • 结构体命名:将Edge的u和v命名更清晰,便于理解。
  • 主函数优化:使用向量存储边,简化了输入处理。
  • 注释增加:添加了注释,解释了代码的主要功能和流程。
  • 递归转换:将递归函数改为迭代式DFS,避免了栈溢出的问题。
  • 异常处理:增加了对输入边界的检查,确保节点数和边数合理。
  • 代码风格统一:调整了代码缩进和格式,使其更易读。
  • 功能模块化:将主要功能分为calculateMinimumCost函数,提高了代码的可维护性。
  • 通过这些优化,代码不仅更清晰易懂,还提高了性能和安全性,适用于处理大规模树结构的场景。

    转载地址:http://town.baihongyu.com/

    你可能感兴趣的文章
    Python块键盘/鼠标输入
    查看>>
    python在使用HTMLTestRunner时,报告为空,错误提示<_io.TextIOWrapper name='<stderr>' mode='w' encoding='utf_8'>...
    查看>>
    python在gpu上运行_【python】python开启GPU加速
    查看>>
    Python在def函数中使用for循环
    查看>>
    Python在def函数中使用for循环
    查看>>
    Python在Conda环境中,但在Windows虚拟环境中没有激活
    查看>>
    python系列【仅供参考】:python pip 错误 ModuleNotFoundError: No module named pip._internal 解决办法
    查看>>
    python图像条状状噪声,使用PYTHON PIL从验证码图像中删除背景嘈杂的线条
    查看>>
    Python图像处理:从内存加载jpeg
    查看>>
    python固定后缀(名物化词汇)词频统计:抽取+统计+可视化
    查看>>
    python商品评论数据采集与分析可视化系统 Flask框架 requests爬虫 NLP情感分析 毕业设计 源码
    查看>>
    python商品数据分析可视化系统(带爬虫)京东销售数据分析 计算机毕业设计 源码下载
    查看>>
    python商品库存管理系统 django框架 商品网站 MySQL数据库 源码下载 计算机毕业设计
    查看>>
    Python哪个版本最稳定好用2023.10.19
    查看>>
    Python和黑客技术的渊源
    查看>>
    Python和RF编写接口自动化
    查看>>
    python和js哪个难_【Python】记一次学习,Python 与 js 在实现闭包时的差异
    查看>>
    python和java哪个更值得学——来自知乎高赞回答
    查看>>
    python命名空间更改_Python模块xml.etree.ElementTree自动修改xml命名空间键
    查看>>
    Python命名中尾随下划线有什么好处?
    查看>>