結果
問題 | No.2396 等差二項展開 |
ユーザー | risujiroh |
提出日時 | 2023-07-29 00:02:59 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 728 ms / 6,000 ms |
コード長 | 3,573 bytes |
コンパイル時間 | 3,733 ms |
コンパイル使用メモリ | 287,140 KB |
実行使用メモリ | 6,948 KB |
最終ジャッジ日時 | 2024-04-16 07:56:42 |
合計ジャッジ時間 | 10,487 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,944 KB |
testcase_02 | AC | 2 ms
6,944 KB |
testcase_03 | AC | 2 ms
6,940 KB |
testcase_04 | AC | 2 ms
6,944 KB |
testcase_05 | AC | 2 ms
6,944 KB |
testcase_06 | AC | 2 ms
6,944 KB |
testcase_07 | AC | 2 ms
6,940 KB |
testcase_08 | AC | 2 ms
6,940 KB |
testcase_09 | AC | 2 ms
6,944 KB |
testcase_10 | AC | 2 ms
6,948 KB |
testcase_11 | AC | 2 ms
6,940 KB |
testcase_12 | AC | 2 ms
6,944 KB |
testcase_13 | AC | 2 ms
6,944 KB |
testcase_14 | AC | 2 ms
6,940 KB |
testcase_15 | AC | 7 ms
6,940 KB |
testcase_16 | AC | 87 ms
6,944 KB |
testcase_17 | AC | 382 ms
6,940 KB |
testcase_18 | AC | 611 ms
6,940 KB |
testcase_19 | AC | 728 ms
6,944 KB |
testcase_20 | AC | 2 ms
6,944 KB |
testcase_21 | AC | 2 ms
6,940 KB |
testcase_22 | AC | 2 ms
6,940 KB |
testcase_23 | AC | 317 ms
6,944 KB |
testcase_24 | AC | 499 ms
6,940 KB |
testcase_25 | AC | 611 ms
6,940 KB |
testcase_26 | AC | 446 ms
6,944 KB |
testcase_27 | AC | 498 ms
6,944 KB |
testcase_28 | AC | 552 ms
6,940 KB |
testcase_29 | AC | 501 ms
6,944 KB |
testcase_30 | AC | 359 ms
6,940 KB |
ソースコード
#if __INCLUDE_LEVEL__ == 0 #include <bits/stdc++.h> using namespace std; #include __BASE_FILE__ namespace { using Mint = atcoder::modint; void solve() { int64_t n; int64_t m; int l; int r; { int b; cin >> tie(n, m, l, r, b); Mint::set_mod(b); } if (l == 1) { print(Mint(1 + m).pow(n)); return; } using Num = vector<Mint>; auto mul = [&](const Num& x, const Num& y) -> Num { Num z(l); for (int i : rep(l)) { for (int j : rep(l)) { int k = i + j; if (l <= k) { k -= l; z[k] += x[i] * y[j] * m; } else { z[k] += x[i] * y[j]; } } } return z; }; auto power = [&](Num a, int64_t n) { assert(0 < n); for (; ~n & 1; n >>= 1) { a = mul(a, a); } Num ret = a; while (n >>= 1) { a = mul(a, a); if (n & 1) { ret = mul(ret, a); } } return ret; }; Num f(l); f[0] = 1; f[1 % l] += 1; f = power(f, n); print(f[r]); } } // namespace int main() { ios::sync_with_stdio(false); cin.tie(nullptr); solve(); } #else // __INCLUDE_LEVEL__ #undef assert #define assert(expr) (expr) || (__builtin_unreachable(), 0) #include <atcoder/modint> template <class T, class U = T> bool chmin(T& x, U&& y) { return y < x && (x = forward<U>(y), true); } template <class T, class U = T> bool chmax(T& x, U&& y) { return x < y && (x = forward<U>(y), true); } namespace std { template <class T1, class T2> istream& operator>>(istream& is, pair<T1, T2>& p) { return is >> p.first >> p.second; } template <class... Ts> istream& operator>>(istream& is, tuple<Ts...>& t) { return apply([&is](auto&... xs) -> istream& { return (is >> ... >> xs); }, t); } template <class... Ts> istream& operator>>(istream& is, tuple<Ts&...>&& t) { return is >> t; } template <class R, enable_if_t<!is_convertible_v<R, string>>* = nullptr> auto operator>>(istream& is, R&& r) -> decltype(is >> *begin(r)) { for (auto&& e : r) { is >> e; } return is; } template <class T1, class T2> ostream& operator<<(ostream& os, const pair<T1, T2>& p) { return os << p.first << ' ' << p.second; } template <class... Ts> ostream& operator<<(ostream& os, const tuple<Ts...>& t) { auto f = [&os](const auto&... xs) -> ostream& { [[maybe_unused]] auto sep = ""; ((os << exchange(sep, " ") << xs), ...); return os; }; return apply(f, t); } template <class R, enable_if_t<!is_convertible_v<R, string_view>>* = nullptr> auto operator<<(ostream& os, R&& r) -> decltype(os << *begin(r)) { auto sep = ""; for (auto&& e : r) { os << exchange(sep, " ") << e; } return os; } } // namespace std namespace atcoder { template <class T, internal::is_modint_t<T>* = nullptr> istream& operator>>(istream& is, T& x) { int v; is >> v; x = T::raw(v); return is; } template <class T, internal::is_modint_t<T>* = nullptr> ostream& operator<<(ostream& os, const T& x) { return os << x.val(); } } // namespace atcoder template <class... Ts> void print(Ts&&... xs) { cout << tie(xs...) << '\n'; } inline auto rep(int l, int r) { return views::iota(min(l, r), r); } inline auto rep(int n) { return rep(0, n); } inline auto rep1(int l, int r) { return rep(l, r + 1); } inline auto rep1(int n) { return rep(1, n + 1); } inline auto per(int l, int r) { return rep(l, r) | views::reverse; } inline auto per(int n) { return per(0, n); } inline auto per1(int l, int r) { return per(l, r + 1); } inline auto per1(int n) { return per(1, n + 1); } #endif // __INCLUDE_LEVEL__