test/yosupo/tree/tree_diameter.test.cpp

• View this file on GitHub
• Last update: 2025-06-10 13:56:45+09:00
• Problem: https://judge.yosupo.jp/problem/tree_diameter

Depends on

• 重み付きグラフ (lib/graph/graph.hpp)
• 木の距離を求める (lib/tree/tree_function.hpp)

Code

// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/tree_diameter
#include <algorithm>
#include <cstdint>
#include <iostream>
#include <vector>
#include "tree/tree_function.hpp"

int main(void) {
    int n;
    std::cin >> n;
    Graph<int> g(n);
    g.input_edges(n - 1, 0);
    auto d = tree_dist<int, std::int64_t>(g, 0);
    int st = std::max_element(d.begin(), d.end()) - d.begin();
    d = tree_dist<int, std::int64_t>(g, st);
    auto p = tree_parent(g, st);
    int gl = std::max_element(d.begin(), d.end()) - d.begin();
    std::vector<int> ans;
    while (gl != -1) {
        ans.emplace_back(gl);
        gl = p[gl];
    }
    std::cout << *std::max_element(d.begin(), d.end()) << ' ' << ans.size() << '\n';
    for (int i = 0; i < (int)ans.size(); ++i)
        std::cout << ans[i] << (i == (int)ans.size() - 1 ? '\n' : ' ');

    return 0;
}

Traceback (most recent call last):
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/resolver.py", line 291, in resolve
    bundled_code = language.bundle(path, basedir=basedir)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus.py", line 242, in bundle
    bundler.update(path)
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus_bundle.py", line 479, in update
    self._resolve(pathlib.Path(included), included_from=path)
  File "/home/runner/.local/lib/python3.12/site-packages/competitive_verifier/oj/verify/languages/cplusplus_bundle.py", line 286, in _resolve
    raise BundleErrorAt(path, -1, "no such header")
competitive_verifier.oj.verify.languages.cplusplus_bundle.BundleErrorAt: tree/tree_function.hpp: line -1: no such header

Test cases

Env	Name	Status	Elapsed	Memory
g++	example_00	AC	3 ms	3 MB
g++	hack_00	AC	2 ms	3 MB
g++	line_00	AC	684 ms	40 MB
g++	max_random_00	AC	753 ms	44 MB
g++	max_random_01	AC	747 ms	44 MB
g++	random_00	AC	534 ms	35 MB
g++	random_01	AC	661 ms	41 MB
g++	random_02	AC	51 ms	8 MB
g++	random_03	AC	610 ms	38 MB
g++	random_04	AC	348 ms	26 MB
g++	small_random_00	AC	3 ms	4 MB
g++	small_random_01	AC	2 ms	3 MB
g++	small_random_02	AC	2 ms	4 MB
g++	small_random_03	AC	2 ms	4 MB
g++	small_random_04	AC	2 ms	4 MB
g++	uni_00	AC	461 ms	47 MB

Back to top page