結果
問題 | No.1392 Don't be together |
ユーザー | pockyny |
提出日時 | 2021-02-13 12:57:47 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 264 ms / 2,000 ms |
コード長 | 1,527 bytes |
コンパイル時間 | 648 ms |
コンパイル使用メモリ | 70,648 KB |
実行使用メモリ | 202,108 KB |
最終ジャッジ日時 | 2023-09-27 21:23:38 |
合計ジャッジ時間 | 7,532 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 173 ms
201,792 KB |
testcase_01 | AC | 169 ms
201,780 KB |
testcase_02 | AC | 169 ms
201,824 KB |
testcase_03 | AC | 161 ms
201,764 KB |
testcase_04 | AC | 152 ms
201,784 KB |
testcase_05 | AC | 154 ms
201,828 KB |
testcase_06 | AC | 230 ms
201,812 KB |
testcase_07 | AC | 191 ms
201,808 KB |
testcase_08 | AC | 222 ms
201,880 KB |
testcase_09 | AC | 264 ms
201,880 KB |
testcase_10 | AC | 180 ms
201,880 KB |
testcase_11 | AC | 192 ms
202,108 KB |
testcase_12 | AC | 175 ms
201,820 KB |
testcase_13 | AC | 188 ms
202,092 KB |
testcase_14 | AC | 187 ms
202,016 KB |
testcase_15 | AC | 182 ms
201,824 KB |
testcase_16 | AC | 192 ms
201,952 KB |
testcase_17 | AC | 179 ms
202,104 KB |
testcase_18 | AC | 191 ms
201,888 KB |
testcase_19 | AC | 186 ms
201,816 KB |
testcase_20 | AC | 169 ms
201,872 KB |
testcase_21 | AC | 160 ms
201,796 KB |
testcase_22 | AC | 174 ms
201,884 KB |
testcase_23 | AC | 172 ms
201,868 KB |
testcase_24 | AC | 162 ms
201,996 KB |
testcase_25 | AC | 171 ms
201,848 KB |
testcase_26 | AC | 165 ms
201,792 KB |
testcase_27 | AC | 162 ms
201,832 KB |
testcase_28 | AC | 171 ms
201,808 KB |
testcase_29 | AC | 165 ms
201,868 KB |
ソースコード
#include <iostream> using namespace std; using ll = long long; const int MX = 100010; ll 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]%mod; } f[0] = fi[0] = 1; for(int i=1;i<MX;i++){ f[i] = f[i-1]*i%mod; fi[i] = fi[i-1]*inv[i]%mod; } } ll nck(ll n, ll k){ if(n<0 || k<0 || n<k) return 0; return f[n]*fi[k]%mod*fi[n-k]%mod; } ll beki[5010][5010]; int par[100010],sz[100010]; void init(int n){ for(int i=0;i<n;i++) par[i] = i,sz[i] = 1; } int find(int x){ if(par[x]==x) return x; return par[x] = find(par[x]); } void unite(int x, int y){ x = find(x); y = find(y); if(x==y) return; if(sz[x]>sz[y]) swap(x,y); par[x] = y; sz[y] += sz[x]; } bool same(int x, int y){ return find(x)==find(y); } int main(){ ll i,j,n,m; cin >> n >> m; init(n); for(i=0;i<n;i++){ int x; cin >> x; x--; unite(i,x); } for(i=0;i<=5000;i++){ beki[i][0] = 1; for(j=1;j<=5000;j++) beki[i][j] = (beki[i][j - 1])*i%mod; } solve(); ll ans = 0; for(i=2;i<=m;i++){ ll s = 1; for(j=0;j<n;j++){ if(find(j)!=j) continue; ll k = 1; if(sz[j]&1) k = mod - 1; (s *= (beki[i - 1][sz[j]] + (i - 1)*k%mod)%mod) %= mod; } (s *= nck(m,i)) %= mod; if((m - i)&1) (ans += mod - s) %= mod; else (ans += s) %= mod; } cout << ans*fi[m]%mod << endl; }