結果

問題 No.2484 Add to Variables
ユーザー MagentorMagentor
提出日時 2023-09-10 00:07:45
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,214 bytes
コンパイル時間 3,801 ms
コンパイル使用メモリ 263,956 KB
実行使用メモリ 8,064 KB
最終ジャッジ日時 2024-07-08 10:24:19
合計ジャッジ時間 5,314 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 3
other AC * 8 WA * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
const int MOD = 998244353;
vector<long long> fact, fact_inv, inv;
void nCk_calc(int s) {
    fact.resize(s + 2);
    fact_inv.resize(s + 2);
    inv.resize(s + 2);
    fact[0] = 1;
    fact[1] = 1;
    fact_inv[0] = fact_inv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < s + 2; i++) {
        fact[i] = fact[i - 1] * i % MOD;
        inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
        fact_inv[i] = fact_inv[i - 1] * inv[i] % MOD;
    }
}
/*  nCk :MODでの二項係数を求める(前処理 int_nCk が必要)
    計算量:O(1)
*/
long long nCk(int n, int k) {
    assert(!(n < k));
    assert(!(n < 0 || k < 0));
    return fact[n] * (fact_inv[k] * fact_inv[n - k] % MOD) % MOD;
}
int main() {
nCk_calc(200005);
 int n;cin>>n;
 int a,b,c,d;cin>>a>>b>>c>>d;
 mint ans = 0;
 for(int i = 0;i <= min(d,b);i++){
   if(b-i>n-d||c<i||n-c<(n-d)-(b-i)||a<b-i||n-a<d-i){break;}
   mint res = 1;
   res *= nCk(n,d);
   res *= nCk(d,i);
   res *= nCk(n-d,b-i);
   res *= nCk(n-(b-i)-(d-i),a-(b-i));
   res *= nCk(n-i-((n-d)-(b-i)),c-i);
   ans += res;
 }
 cout << ans.val() << endl;return 0;
}
0