結果
問題 | No.2381 Gift Exchange Party |
ユーザー | cho435 |
提出日時 | 2023-07-14 22:00:32 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 830 bytes |
コンパイル時間 | 4,304 ms |
コンパイル使用メモリ | 262,768 KB |
実行使用メモリ | 81,476 KB |
最終ジャッジ日時 | 2023-10-14 12:10:18 |
合計ジャッジ時間 | 11,125 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 180 ms
81,008 KB |
testcase_01 | AC | 180 ms
81,212 KB |
testcase_02 | AC | 190 ms
81,208 KB |
testcase_03 | AC | 182 ms
81,072 KB |
testcase_04 | AC | 190 ms
81,020 KB |
testcase_05 | AC | 186 ms
81,016 KB |
testcase_06 | AC | 174 ms
81,096 KB |
testcase_07 | AC | 180 ms
80,960 KB |
testcase_08 | AC | 180 ms
80,972 KB |
testcase_09 | AC | 176 ms
81,208 KB |
testcase_10 | AC | 173 ms
81,092 KB |
testcase_11 | AC | 174 ms
80,964 KB |
testcase_12 | AC | 179 ms
81,156 KB |
testcase_13 | AC | 173 ms
81,156 KB |
testcase_14 | AC | 177 ms
80,960 KB |
testcase_15 | AC | 176 ms
81,004 KB |
testcase_16 | AC | 185 ms
81,212 KB |
testcase_17 | AC | 184 ms
81,212 KB |
testcase_18 | AC | 180 ms
80,964 KB |
testcase_19 | AC | 174 ms
81,088 KB |
testcase_20 | AC | 195 ms
80,972 KB |
testcase_21 | RE | - |
testcase_22 | AC | 203 ms
81,104 KB |
testcase_23 | AC | 176 ms
81,104 KB |
testcase_24 | AC | 168 ms
80,996 KB |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using ll = long long; #define rep(i,n) for(int i=0;i<(int)(n);i++) using mint = atcoder::modint998244353; // combination MOD const int MOD=998244353; const int MAX_N=5e6; vector<mint> fac(MAX_N+1,1); vector<mint> finv(MAX_N+1,1); vector<ll> inv(MAX_N+1,1); void comb_setup(){ for(int i=2;i<=MAX_N;i++){ fac.at(i)=(fac.at(i-1)*i); inv.at(i)=MOD-(inv.at(MOD%i)*(MOD/i))%MOD; finv.at(i)=(finv.at(i-1)*inv.at(i)); } } mint comb(ll n,ll k){ if(n<k) return 0; return fac.at(n)*finv.at(n-k)*finv.at(k); } int main(){ comb_setup(); ll n,p; cin>>n>>p; mint ans=fac.at(n); for(ll i=0;i*p<=n;i++){ mint tmp=comb(n,i*p); tmp*=fac.at(i*p); tmp*=finv.at(p).pow(i); tmp*=finv.at(i); tmp*=fac.at(p-1).pow(i); ans-=tmp; } cout<<ans.val()<<endl; }