Test/CF/ECR157-F.test.cpp

• View this file on GitHub
• Last update: 2024-06-25 21:03:43+09:00
• Problem: https://onlinejudge.u-aizu.ac.jp/courses/lesson/2/ITP1/1/ITP1_1_A
• Link: https://codeforces.com/contest/1895/submission/255171867

Depends on

• Src/Algebra/Ring/UsualRing.hpp
• Src/LinearAlgebra/Matrix.hpp
• ioまわりの設定 (Src/Template/IOSetting.hpp)
• 標準データ型のエイリアス (Src/Template/TypeAlias.hpp)

Code

#define PROBLEM "https://onlinejudge.u-aizu.ac.jp/courses/lesson/2/ITP1/1/ITP1_1_A"

#include "../../Src/Template/IOSetting.hpp"
#include "../../Src/Algebra/Ring/UsualRing.hpp"
#include "../../Src/LinearAlgebra/Matrix.hpp"
#include "atcoder/modint.hpp"

#include <iostream>

using namespace zawa;
using mint = atcoder::modint1000000007;
using M = Matrix<UsualRing<mint>>;

/*
 * ECR157-F Fancy Arrays
 * https://codeforces.com/contest/1895/submission/255171867
 */

mint solve() {
    int n, x, k; std::cin >> n >> x >> k;
    mint ans{};
    if (x > 0) {
        M A(x);
        for (int i{} ; i < x ; i++) {
            for (int j{} ; j < x ; j++) {
                if (std::abs(i - j) <= k) {
                    A[i][j] = mint::raw(1);
                }
            }
        }
        A = A.pow(n - 1);
        for (int i{} ; i < x ; i++) {
            for (int j{} ; j < x ; j++) {
                ans += A[i][j];
            }
        }
    }
    ans = mint{x + k} * mint{2 * k + 1}.pow(n - 1) - ans;
    return ans;
}

int main() {
#ifdef ONLINE_JUDGE
    SetFastIO();
    int t; std::cin >> t;
    while (t--) {
        std::cout << solve().val() << '\n';
    }
#else
    std::cout << "Hello World" << '\n';
#endif
}

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.13.5/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.5/x64/lib/python3.13/site-packages/onlinejudge_verify/languages/cplusplus.py", line 187, in bundle
    bundler.update(path)
    ~~~~~~~~~~~~~~^^^^^^
  File "/opt/hostedtoolcache/Python/3.13.5/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.5/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.hpp: line -1: no such header

Back to top page