結果
問題 | No.1621 Sequence Inversions |
ユーザー | neterukun |
提出日時 | 2021-07-22 22:54:53 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,064 bytes |
コンパイル時間 | 1,593 ms |
コンパイル使用メモリ | 86,816 KB |
実行使用メモリ | 77,884 KB |
最終ジャッジ日時 | 2023-09-24 18:36:30 |
合計ジャッジ時間 | 5,614 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 76 ms
71,360 KB |
testcase_01 | AC | 74 ms
71,308 KB |
testcase_02 | AC | 85 ms
76,424 KB |
testcase_03 | AC | 75 ms
71,428 KB |
testcase_04 | AC | 81 ms
76,364 KB |
testcase_05 | AC | 74 ms
71,128 KB |
testcase_06 | AC | 81 ms
76,036 KB |
testcase_07 | AC | 88 ms
76,156 KB |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | AC | 116 ms
77,372 KB |
testcase_11 | WA | - |
testcase_12 | AC | 299 ms
77,300 KB |
testcase_13 | AC | 540 ms
77,884 KB |
testcase_14 | WA | - |
testcase_15 | AC | 140 ms
76,264 KB |
testcase_16 | AC | 167 ms
76,992 KB |
testcase_17 | AC | 236 ms
77,136 KB |
testcase_18 | AC | 159 ms
76,652 KB |
testcase_19 | AC | 85 ms
76,468 KB |
testcase_20 | AC | 121 ms
76,308 KB |
testcase_21 | AC | 256 ms
77,060 KB |
testcase_22 | AC | 128 ms
76,468 KB |
testcase_23 | AC | 294 ms
77,032 KB |
testcase_24 | AC | 74 ms
71,192 KB |
testcase_25 | AC | 80 ms
76,508 KB |
testcase_26 | AC | 73 ms
71,308 KB |
testcase_27 | WA | - |
testcase_28 | WA | - |
ソースコード
class Combination: def __init__(self, n, MOD): self.f = [1] for i in range(1, n + 1): self.f.append(self.f[-1] * i % MOD) self.inv_f = [0] * (n + 1) self.inv_f[n] = pow(self.f[n], MOD - 2, MOD) for i in reversed(range(n)): self.inv_f[i] = self.inv_f[i + 1] * (i + 1) % MOD self.MOD = MOD def inv(self, k): """get inverse(k)""" return (self.inv_f[k] * self.f[k - 1]) % self.MOD def fact(self, k): """get k!""" return self.f[k] def inv_fact(self, k): """get inverse(k!)""" return self.inv_f[k] def perm(self, k, r): """get kPr""" if k < r: return 0 return (self.f[k] * self.inv_f[k - r]) % self.MOD def comb(self, k, r): """get kCr""" if k < r: return 0 return (self.f[k] * self.inv_f[k - r] * self.inv_f[r]) % self.MOD def combination(k, r, MOD): """kCr O(r)""" if k < r: return 0 r = min(r, k - r) numer, denom = 1, 1 for l in range(r): numer *= (k - l) numer %= MOD denom *= l + 1 denom %= MOD return numer * pow(denom, MOD - 2, MOD) % MOD def encoding(s): n = len(s) begin, cnt = 0, 0 ans = [] if n == 0: return ans for end in range(n + 1): if end == n or s[begin] != s[end]: ans.append((s[begin], cnt)) begin, cnt = end, 1 else: cnt += 1 return ans n, k = map(int, input().split()) a = list(map(int, input().split())) MOD = 998244353 comb = Combination(1000, MOD) a = sorted(a) comp = encoding(a) cnts = [cnt for _, cnt in comp] dp = [0] * (k + 1) dp[0] = 1 ru_cnt = 0 for cnt in cnts: dq = [0] * (k + 1) for count in range(k + 1): for j in range(ru_cnt + 1): if count + j < k + 1: dq[count + j] += dp[count] * comb.comb(j + cnt - 1, cnt - 1) dq[count + j] %= MOD ru_cnt += cnt dp, dq = dq, dp print(dp[-1] % MOD)