結果
問題 | No.665 Bernoulli Bernoulli |
ユーザー | rlangevin |
提出日時 | 2023-09-21 00:08:33 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,023 bytes |
コンパイル時間 | 174 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 76,800 KB |
最終ジャッジ日時 | 2024-07-06 18:55:09 |
合計ジャッジ時間 | 40,823 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 40 ms
58,752 KB |
testcase_01 | AC | 44 ms
58,496 KB |
testcase_02 | TLE | - |
testcase_03 | TLE | - |
testcase_04 | TLE | - |
testcase_05 | TLE | - |
testcase_06 | TLE | - |
testcase_07 | TLE | - |
testcase_08 | TLE | - |
testcase_09 | TLE | - |
testcase_10 | TLE | - |
testcase_11 | TLE | - |
testcase_12 | TLE | - |
testcase_13 | TLE | - |
testcase_14 | TLE | - |
testcase_15 | TLE | - |
testcase_16 | TLE | - |
testcase_17 | TLE | - |
testcase_18 | TLE | - |
ソースコード
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 + 1) B[0] = 1 B[1] = (- pow(2, mod - 2, mod)) % mod for i in range(2, n, 2): inv = invfact[i+1] * fact[i] temp = - inv * comb(i + 1, 1) * B[1] for k in range(0, i, 2): temp -= inv * comb(i + 1, k) * B[k] temp %= mod B[i] = temp 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)