結果
問題 | No.2058 Binary String |
ユーザー |
|
提出日時 | 2022-08-26 22:03:53 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 321 ms / 2,000 ms |
コード長 | 1,095 bytes |
コンパイル時間 | 270 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 99,328 KB |
最終ジャッジ日時 | 2024-10-13 22:39:58 |
合計ジャッジ時間 | 7,669 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
class Combination: def __init__(self, MX=10**6, MOD=998244353): self.MX = MX self.MOD = MOD self.fact = [1] * (MX + 1) self.inv = [1] * (MX + 1) self.f_inv = [1] * (MX + 1) for i in range(2, MX + 1): self.fact[i] = (self.fact[i - 1] * i) % MOD self.inv[i] = ( - (MOD // i) * (self.inv[MOD % i])) % MOD self.f_inv[i] = (self.f_inv[i - 1] * self.inv[i]) % MOD def invs(self, n): if n <= self.MX: return self.inv[n] else: return pow(n, self.MOD - 2, self.MOD) def f(self, n): return self.fact[n] def p(self, n, r): if r > n or r < 0: return 0 return (self.fact[n] * self.f_inv[r]) % self.MOD def c(self, n, r): if r > n or r < 0: return 0 return (self.fact[n] * self.f_inv[r] * self.f_inv[n - r]) % self.MOD MOD = 998244353 N, K = map(int, input().split()) com = Combination() ans = 0 for i in range(N): ans = (ans + com.c(N - 1, i) * pow(i, K, MOD)) % MOD print(ans)