結果
| 問題 |
No.2058 Binary String
|
| コンテスト | |
| ユーザー |
rlangevin
|
| 提出日時 | 2023-10-09 13:08:36 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 162 ms / 2,000 ms |
| コード長 | 510 bytes |
| コンパイル時間 | 1,118 ms |
| コンパイル使用メモリ | 82,400 KB |
| 実行使用メモリ | 79,216 KB |
| 最終ジャッジ日時 | 2024-07-26 18:30:47 |
| 合計ジャッジ時間 | 3,832 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 23 |
ソースコード
N, K = map(int, input().split())
mod = 998244353
n = N
fact = [1] * (n + 1)
invfact = [1] * (n + 1)
for i in range(1, n):
fact[i + 1] = ((i+1) * fact[i]) % mod
invfact[n] = pow(fact[n], mod - 2, mod)
for i in range(n - 1, -1, -1):
invfact[i] = invfact[i + 1] * (i + 1) % mod
def comb(n, r):
if n < 0 or r < 0 or n - r < 0:
return 0
return fact[n] * invfact[r] * invfact[n - r] % mod
ans = 0
for i in range(N):
ans += pow(i, K, mod) * comb(N - 1, i)
ans %= mod
print(ans)
rlangevin