test/yosupo/enumerate_triangles.test.cpp

View this file on GitHub
Last update: 2025-05-24 15:06:07+09:00
Problem: https://judge.yosupo.jp/problem/enumerate_triangles

Depends on

Enuemrate $C_3$, Count $C_4, K_4$ (src/graph/count_graphs.hpp)

Code

#define PROBLEM "https://judge.yosupo.jp/problem/enumerate_triangles"

#include <iostream>
#include "atcoder/modint"
#include "graph/count_graphs.hpp"

using mint = atcoder::modint998244353;

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int N, M;
    std::cin >> N >> M;
    std::vector<int> x(N);
    for (int i = 0; i < N; i++) std::cin >> x[i];
    std::vector<std::vector<int>> G(N);
    for (; M--;) {
        int u, v;
        std::cin >> u >> v;
        G[u].emplace_back(v);
        G[v].emplace_back(u);
    }

    mint ans = 0;
    for (auto [a, b, c] : count_graphs::enumerate_C3(G)) ans += mint(1) * x[a] * x[b] * x[c];
    std::cout << ans.val() << '\n';
    return 0;
}

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.13.3/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.3/x64/lib/python3.13/site-packages/onlinejudge_verify/languages/cplusplus.py", line 187, in bundle
    bundler.update(path)
    ~~~~~~~~~~~~~~^^^^^^
  File "/opt/hostedtoolcache/Python/3.13.3/x64/lib/python3.13/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 401, in update
    self.update(self._resolve(pathlib.Path(included), included_from=path))
                ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.13.3/x64/lib/python3.13/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 260, in _resolve
    raise BundleErrorAt(path, -1, "no such header")
onlinejudge_verify.languages.cplusplus_bundle.BundleErrorAt: atcoder/modint: line -1: no such header