結果
問題 | 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 = MXself.MOD = MODself.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) % MODself.inv[i] = ( - (MOD // i) * (self.inv[MOD % i])) % MODself.f_inv[i] = (self.f_inv[i - 1] * self.inv[i]) % MODdef 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 0return (self.fact[n] * self.f_inv[r]) % self.MODdef c(self, n, r):if r > n or r < 0:return 0return (self.fact[n] * self.f_inv[r] * self.f_inv[n - r]) % self.MODMOD = 998244353N, K = map(int, input().split())com = Combination()ans = 0for i in range(N):ans = (ans + com.c(N - 1, i) * pow(i, K, MOD)) % MODprint(ans)