tsp/mst_edges.hpp

View this file on GitHub
Last update: 2023-02-16 23:07:28+09:00
Include: #include "tsp/mst_edges.hpp"

Code

#pragma once

#include <algorithm>
#include <limits>
#include <utility>
#include <vector>

template <class DistanceMatrix>
std::vector<std::pair<int, int>> mst_edges(const DistanceMatrix &dist) {
    using T = decltype((*dist.adjacents(0).begin()).second);

    if (dist.n() <= 1) return {};

    std::vector<T> dp(dist.n(), std::numeric_limits<T>::max());
    std::vector<int> prv(dist.n(), -1);
    std::vector<int> used(dist.n());
    std::vector<std::pair<int, int>> ret(dist.n() - 1);

    for (int t = 0; t < dist.n(); ++t) {
        int x = std::min_element(dp.cbegin(), dp.cend()) - dp.cbegin();
        dp.at(x) = std::numeric_limits<T>::max();
        used.at(x) = 1;
        if (t > 0) ret.at(t - 1) = {prv.at(x), x};

        for (auto [y, len] : dist.adjacents(x)) {
            if (!used.at(y) and len < dp.at(y)) dp.at(y) = len, prv.at(y) = x;
        }
    }

    return ret;
}

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.13.7/x64/lib/python3.13/site-packages/onlinejudge_verify/documentation/build.py", line 71, in _render_source_code_stat
    bundled_code = language.bundle(stat.path, basedir=basedir, options={'include_paths': [basedir]}).decode()
                   ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.13.7/x64/lib/python3.13/site-packages/onlinejudge_verify/languages/cplusplus.py", line 187, in bundle
    bundler.update(path)
    ~~~~~~~~~~~~~~^^^^^^
  File "/opt/hostedtoolcache/Python/3.13.7/x64/lib/python3.13/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 312, in update
    raise BundleErrorAt(path, i + 1, "#pragma once found in a non-first line")
onlinejudge_verify.languages.cplusplus_bundle.BundleErrorAt: tsp/mst_edges.hpp: line 2: #pragma once found in a non-first line