結果
問題 | No.1191 数え上げを愛したい(数列編) |
ユーザー | hiro71687k |
提出日時 | 2023-04-10 16:49:54 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 147 ms / 2,000 ms |
コード長 | 1,635 bytes |
コンパイル時間 | 5,150 ms |
コンパイル使用メモリ | 264,008 KB |
実行使用メモリ | 34,208 KB |
最終ジャッジ日時 | 2024-04-15 21:29:23 |
合計ジャッジ時間 | 8,207 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 103 ms
29,048 KB |
testcase_01 | AC | 142 ms
33,228 KB |
testcase_02 | AC | 143 ms
33,484 KB |
testcase_03 | AC | 99 ms
29,432 KB |
testcase_04 | AC | 140 ms
33,820 KB |
testcase_05 | AC | 93 ms
25,488 KB |
testcase_06 | AC | 145 ms
33,744 KB |
testcase_07 | AC | 139 ms
33,576 KB |
testcase_08 | AC | 138 ms
33,708 KB |
testcase_09 | AC | 139 ms
33,528 KB |
testcase_10 | AC | 142 ms
34,208 KB |
testcase_11 | AC | 145 ms
34,096 KB |
testcase_12 | AC | 146 ms
33,764 KB |
testcase_13 | AC | 147 ms
33,704 KB |
testcase_14 | AC | 147 ms
33,988 KB |
testcase_15 | AC | 59 ms
19,944 KB |
testcase_16 | AC | 59 ms
20,048 KB |
testcase_17 | AC | 69 ms
21,488 KB |
testcase_18 | AC | 72 ms
21,744 KB |
testcase_19 | AC | 62 ms
20,420 KB |
testcase_20 | AC | 71 ms
21,952 KB |
testcase_21 | AC | 54 ms
19,024 KB |
testcase_22 | AC | 99 ms
29,060 KB |
testcase_23 | AC | 48 ms
17,792 KB |
testcase_24 | AC | 48 ms
17,988 KB |
testcase_25 | AC | 47 ms
17,864 KB |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using namespace atcoder; using ll=long long; using ld=long double; ld pie=3.141592653589793; ll inf=1444999999994; ll mod=998244353; const int MAX = 610000; const int MOD = mod; long long fac[MAX], finv[MAX], inv[MAX]; // テーブルを作る前処理 void COMinit() { fac[0] = fac[1] = 1; finv[0] = finv[1] = 1; inv[1] = 1; for (int i = 2; i < MAX; i++){ fac[i] = fac[i - 1] * i % MOD; inv[i] = MOD - inv[MOD%i] * (MOD / i) % MOD; finv[i] = finv[i - 1] * inv[i] % MOD; } } // 二項係数計算 long long COM(int n, int k){ if (n < k) return 0; if (n < 0 || k < 0) return 0; return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD; } int main(){ ll n,m,a,b; cin >> n >> m >> a >> b; vector<ll>x(m,0); COMinit(); for (ll i = n-1; i >=0; i--) { if (i*a+(n-1-i)*(b+1)>=m) { break; } ll y=i*a+(n-1-i)*(b+1); x[y]=COM(n-1,i); if ((n-1-i)%2) { x[y]*=-1; x[y]+=mod; x[y]%=mod; } } vector<ll>y(m+1,0); for (ll i = 0; i <=m; i++) { y[i]=COM(n-2+i,n-2); } vector<ll>z=convolution<998244353>(x,y); ll ans=0; vector<ll>kaijo(n+100,1); for (ll i = 1; i < kaijo.size(); i++) { kaijo[i]=kaijo[i-1]*i; kaijo[i]%=mod; } for (ll i = 0; i <m; i++) { if (i>b) { continue; } ans+=(((z[i]*(m-i))%mod)*kaijo[n])%mod; ans%=mod; } cout << ans << endl; }