結果
問題 | No.2484 Add to Variables |
ユーザー |
![]() |
提出日時 | 2023-10-03 23:32:06 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 160 ms / 2,000 ms |
コード長 | 1,118 bytes |
コンパイル時間 | 2,134 ms |
コンパイル使用メモリ | 112,264 KB |
実行使用メモリ | 29,052 KB |
最終ジャッジ日時 | 2024-07-26 14:14:03 |
合計ジャッジ時間 | 6,830 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 39 |
ソースコード
#include <iostream> #include <vector> #include <atcoder/modint> #include <atcoder/convolution> using namespace std; using namespace atcoder; using ll = long long; using mint = modint998244353; const int MX = 1000010; mint f[MX],inv[MX],fi[MX]; constexpr ll mod = 998244353; void solve(){ inv[1] = 1; for(int i=2;i<MX;i++){ inv[i] = mod - (mod/i)*inv[mod%i]; } f[0] = fi[0] = 1; for(int i=1;i<MX;i++){ f[i] = f[i-1]*i; fi[i] = fi[i-1]*inv[i]; } } int a[200010]; int main(){ solve(); int i,n,m; cin >> n >> m; for(i=0;i<n;i++) cin >> a[i]; vector<mint> ff(m + 1),gg(m + 1),hh(m + 1); for(i=0;i<=m;i++){ if(a[0] - a[1] + i>=0) ff[i] = fi[i]*fi[a[0] - a[1] + i]; if(a[1] - a[2] + i>=0) gg[i] = fi[i]*fi[a[1] - a[2] + i]; if(a[2] - a[3] + i>=0) hh[i] = fi[i]*fi[a[2] - a[3] + i]; } ff = convolution(ff,gg); ff = convolution(ff,hh); mint ans = 0; for(i=0;i<=m;i++){ if(a[3] - i>=0 && (m - a[0]) - i>=0) ans += ff[i]*fi[a[3] - i]*fi[m - a[0] - i]; } ans *= f[m]; cout << ans.val() << "\n"; }