結果
問題 | No.665 Bernoulli Bernoulli |
ユーザー | rlangevin |
提出日時 | 2023-09-20 12:45:47 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 901 bytes |
コンパイル時間 | 126 ms |
コンパイル使用メモリ | 81,952 KB |
実行使用メモリ | 83,304 KB |
最終ジャッジ日時 | 2024-07-06 08:04:23 |
合計ジャッジ時間 | 3,718 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 36 ms
67,240 KB |
testcase_01 | AC | 34 ms
58,836 KB |
testcase_02 | TLE | - |
testcase_03 | -- | - |
testcase_04 | -- | - |
testcase_05 | -- | - |
testcase_06 | -- | - |
testcase_07 | -- | - |
testcase_08 | -- | - |
testcase_09 | -- | - |
testcase_10 | -- | - |
testcase_11 | -- | - |
testcase_12 | -- | - |
testcase_13 | -- | - |
testcase_14 | -- | - |
testcase_15 | -- | - |
testcase_16 | -- | - |
testcase_17 | -- | - |
testcase_18 | -- | - |
ソースコード
n = 10005 mod = 10 ** 9 + 7 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 def bernoulli(n, mod): B = [0] * n B[0] = 1 for i in range(1, n): inv = invfact[i+1] * fact[i] for k in range(i): B[i] -= inv * comb(i + 1, k) * B[k] B[i] %= mod return B N, K = map(int, input().split()) ans = 0 v = pow(N + 1, K + 1, mod) inv = pow(N + 1, mod - 2, mod) B = bernoulli(K + 5, mod) v2 = fact[K + 1] for i in range(K + 1): ans += comb(K + 1, i) * B[i] * v v *= inv ans %= mod v %= mod ans *= pow(K + 1, mod - 2, mod) print(ans%mod)