verify/unit_test/binom_table.test.cpp

• View this file on GitHub
• Last update: 2025-06-04 12:11:29+09:00
• Problem: https://judge.yosupo.jp/problem/aplusb

Depends on

• math_mod/binom_table.hpp
• math_mod/comb.hpp
• modint/modint.hpp
• random/gen.hpp
• random/seed.hpp
• template/constant.hpp
• template/fastio.hpp
• template/io_util.hpp
• template/macros.hpp
• template/template.hpp
• template/type_alias.hpp
• type_traits/integral.hpp
• type_traits/io.hpp

Code

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

#include "../../math_mod/binom_table.hpp"
#include "../../math_mod/comb.hpp"
#include "../../modint/modint.hpp"
#include "../../random/gen.hpp"
#include "../../template/template.hpp"
using namespace std;

int main() {
    int a, b;
    kin >> a >> b;
    kout << a + b << kendl;

    // aliasの定義
    using mint = kk2::mint998;
    using BTable = kk2::BinomTable<mint>;
    using Comb = kk2::Comb<mint>;

    // binom_tableのテスト
    rep (1000) {
        int max_n = kk2::random::rng(1, 1000);
        BTable::set_upper(max_n);
        
        rep (1000) {
            int n = kk2::random::rng(0, max_n + 1);
            int k = kk2::random::rng(-10, n + 10);
            
            auto table_result = BTable::binom(n, k);
            auto comb_result = Comb::binom(n, k);
            
            if (table_result != comb_result) {
                cerr << "binom mismatch: n=" << n << ", k=" << k << endl;
                cerr << "table: " << table_result << ", comb: " << comb_result << endl;
                exit(1);
            }
        }
        
        // エッジケースのテスト
        if (BTable::binom(0, 0) != 1) {
            cerr << "binom(0,0) should be 1" << endl;
            exit(1);
        }
        if (BTable::binom(max_n, 0) != 1) {
            cerr << "binom(n,0) should be 1" << endl;
            exit(1);
        }
        if (BTable::binom(max_n, max_n) != 1) {
            cerr << "binom(n,n) should be 1" << endl;
            exit(1);
        }
        if (BTable::binom(max_n, -1) != 0) {
            cerr << "binom(n,-1) should be 0" << endl;
            exit(1);
        }
        if (BTable::binom(max_n, max_n + 1) != 0) {
            cerr << "binom(n,n+1) should be 0" << endl;
            exit(1);
        }
    }

    return 0;
}

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/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.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus.py", line 187, in bundle
    bundler.update(path)
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/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.12.0/x64/lib/python3.12/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.12.0/x64/lib/python3.12/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: type_traits/integral.hpp: line 4: #pragma once found in a non-first line

Back to top page