結果
問題 | No.2522 Fall in love, Girls! |
ユーザー |
|
提出日時 | 2023-10-27 22:36:48 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 125 ms / 2,000 ms |
コード長 | 2,199 bytes |
コンパイル時間 | 971 ms |
コンパイル使用メモリ | 84,048 KB |
実行使用メモリ | 31,040 KB |
最終ジャッジ日時 | 2024-09-25 14:39:48 |
合計ジャッジ時間 | 3,772 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 4 ms
11,720 KB |
testcase_01 | AC | 5 ms
11,684 KB |
testcase_02 | AC | 5 ms
11,676 KB |
testcase_03 | AC | 103 ms
11,552 KB |
testcase_04 | AC | 5 ms
11,720 KB |
testcase_05 | AC | 5 ms
11,524 KB |
testcase_06 | AC | 9 ms
13,256 KB |
testcase_07 | AC | 14 ms
16,248 KB |
testcase_08 | AC | 16 ms
16,924 KB |
testcase_09 | AC | 18 ms
19,292 KB |
testcase_10 | AC | 124 ms
31,036 KB |
testcase_11 | AC | 105 ms
31,040 KB |
testcase_12 | AC | 19 ms
19,216 KB |
testcase_13 | AC | 115 ms
30,908 KB |
testcase_14 | AC | 124 ms
30,896 KB |
testcase_15 | AC | 125 ms
31,020 KB |
testcase_16 | AC | 121 ms
30,896 KB |
testcase_17 | AC | 39 ms
30,664 KB |
testcase_18 | AC | 36 ms
30,792 KB |
testcase_19 | AC | 39 ms
30,700 KB |
testcase_20 | AC | 34 ms
30,204 KB |
testcase_21 | AC | 33 ms
28,984 KB |
testcase_22 | AC | 32 ms
29,780 KB |
testcase_23 | AC | 36 ms
30,132 KB |
testcase_24 | AC | 32 ms
25,596 KB |
testcase_25 | AC | 106 ms
26,948 KB |
testcase_26 | AC | 101 ms
21,960 KB |
testcase_27 | AC | 19 ms
19,848 KB |
testcase_28 | AC | 93 ms
13,228 KB |
testcase_29 | AC | 26 ms
24,520 KB |
testcase_30 | AC | 38 ms
29,680 KB |
testcase_31 | AC | 27 ms
24,576 KB |
testcase_32 | AC | 95 ms
20,828 KB |
testcase_33 | AC | 31 ms
27,972 KB |
ソースコード
#include<iostream> #include<vector> #include<algorithm> #include<cassert> #include<atcoder/modint> using namespace std; #include<vector> template<typename T> struct combination{ vector<T>fac,ifac; combination(size_t N=0):fac(1,1),ifac(1,1) { make_table(N); } void make_table(size_t N) { if(fac.size()>N)return; size_t now=fac.size(); N=max(N,now*2); fac.resize(N+1); ifac.resize(N+1); for(size_t i=now;i<=N;i++)fac[i]=fac[i-1]*i; ifac[N]=1/fac[N]; for(size_t i=N;i-->now;)ifac[i]=ifac[i+1]*(i+1); } T factorial(size_t n) { make_table(n); return fac[n]; } T invfac(size_t n) { make_table(n); return ifac[n]; } T P(size_t n,size_t k) { if(n<k)return 0; make_table(n); return fac[n]*ifac[n-k]; } T C(size_t n,size_t k) { if(n<k)return 0; make_table(n); return fac[n]*ifac[n-k]*ifac[k]; } T H(size_t n,size_t k) { if(n==0)return k==0?1:0; return C(n-1+k,k); } }; using mint=atcoder::modint998244353; combination<mint>C; int N,M,K; int X[380],Y[380]; mint dp[1<<20][2]; int need[20]; int main() { ios::sync_with_stdio(false); cin.tie(nullptr); cin>>N>>M>>K; vector<int>vs; vs.reserve(20); for(int i=0;i<K;i++) { cin>>X[i]>>Y[i]; vs.push_back(X[i]); vs.push_back(Y[i]); } sort(vs.begin(),vs.end()); vs.erase(unique(vs.begin(),vs.end()),vs.end()); assert(vs.size()<=20); if(vs.empty()) { cout<<(C.P(N-1,M)*C.factorial(N-M)).val()<<endl; return 0; } for(int i=0;i<K;i++) { int u=lower_bound(vs.begin(),vs.end(),X[i])-vs.begin(); int v=lower_bound(vs.begin(),vs.end(),Y[i])-vs.begin(); need[v]|=1<<u; } for(int i=0;i<vs.size();i++)if(need[i]==0)dp[1<<i][vs[i]<=M]=1; for(int i=0;i<1<<vs.size();i++)for(int j=0;j<vs.size();j++)if(!(i>>j&1)&&(need[j]&i)==need[j]) { dp[i|1<<j][0]+=dp[i][0]; dp[i|1<<j][1]+=dp[i][1]; } mint ans=C.C(N,vs.size())*C.factorial(N-vs.size())*(dp[(1<<vs.size())-1][0]+dp[(1<<vs.size())-1][1]); {//1 ans-=C.C(N-1,vs.size()-1)*C.factorial(N-vs.size())*dp[(1<<vs.size())-1][1]; } { int r=M; for(int v:vs)if(v<=M)r--; if(r>=1) { ans-=C.C(N-1,vs.size())*r*C.factorial(N-vs.size()-1)*(dp[(1<<vs.size())-1][0]+dp[(1<<vs.size())-1][1]); } } cout<<ans.val()<<endl; }