結果
問題 | No.1191 数え上げを愛したい(数列編) |
ユーザー |
![]() |
提出日時 | 2020-08-22 15:02:52 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 60 ms / 2,000 ms |
コード長 | 1,251 bytes |
コンパイル時間 | 1,747 ms |
コンパイル使用メモリ | 171,032 KB |
実行使用メモリ | 11,584 KB |
最終ジャッジ日時 | 2024-10-15 09:19:57 |
合計ジャッジ時間 | 2,966 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 26 |
ソースコード
#include <bits/stdc++.h> using namespace std; const long long MOD = 998244353; long long modpow(long long a, long long b){ long long ans = 1; while (b > 0){ if (b % 2 == 1){ ans *= a; ans %= MOD; } a *= a; a %= MOD; b /= 2; } return ans; } long long modinv(long long a){ return modpow(a, MOD - 2); } vector<long long> mf = {1}; vector<long long> mfi = {1}; long long modfact(int n){ if (mf.size() > n){ return mf[n]; } else { for (int i = mf.size(); i <= n; i++){ long long next = mf.back() * i % MOD; mf.push_back(next); mfi.push_back(modinv(next)); } return mf[n]; } } long long modfactinv(int n){ if (mfi.size() > n){ return mfi[n]; } else { return modinv(modfact(n)); } } long long modbinom(int n, int k){ if (n < 0 || k < 0 || k > n){ return 0; } else { return modfact(n) * modfactinv(k) % MOD * modfactinv(n - k) % MOD; } } int main(){ int N, M, A, B; cin >> N >> M >> A >> B; if ((long long) A * (N - 1) > B){ cout << 0 << endl; } else { B -= A * (N - 1); long long ans = 0; for (int i = 0; i <= B; i++){ ans += modbinom(N - 2 + i, i) * (M - i - A * (N - 1)) % MOD; ans %= MOD; } ans *= modfact(N); ans %= MOD; cout << ans << endl; } }