結果

問題 No.665 Bernoulli Bernoulli
ユーザー mkawa2mkawa2
提出日時 2021-02-11 17:05:10
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 362 ms / 2,000 ms
コード長 1,440 bytes
コンパイル時間 574 ms
コンパイル使用メモリ 87,068 KB
実行使用メモリ 78,468 KB
最終ジャッジ日時 2023-09-25 06:46:31
合計ジャッジ時間 8,035 ms
ジャッジサーバーID
(参考情報)
judge11 / judge13
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 92 ms
76,112 KB
testcase_01 AC 92 ms
76,256 KB
testcase_02 AC 362 ms
77,396 KB
testcase_03 AC 362 ms
78,288 KB
testcase_04 AC 345 ms
78,160 KB
testcase_05 AC 327 ms
78,252 KB
testcase_06 AC 326 ms
77,928 KB
testcase_07 AC 318 ms
78,108 KB
testcase_08 AC 314 ms
78,468 KB
testcase_09 AC 348 ms
78,124 KB
testcase_10 AC 320 ms
78,272 KB
testcase_11 AC 348 ms
78,272 KB
testcase_12 AC 345 ms
78,172 KB
testcase_13 AC 351 ms
78,364 KB
testcase_14 AC 356 ms
78,324 KB
testcase_15 AC 320 ms
78,296 KB
testcase_16 AC 328 ms
78,300 KB
testcase_17 AC 323 ms
77,928 KB
testcase_18 AC 315 ms
78,260 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys

sys.setrecursionlimit(10**6)
int1 = lambda x: int(x)-1
p2D = lambda x: print(*x, sep="\n")
def II(): return int(sys.stdin.buffer.readline())
def MI(): return map(int, sys.stdin.buffer.readline().split())
def MI1(): return map(int1, sys.stdin.buffer.readline().split())
def LI(): return list(map(int, sys.stdin.buffer.readline().split()))
def LI1(): return list(map(int1, sys.stdin.buffer.readline().split()))
def LLI(rows_number): return [LI() for _ in range(rows_number)]
def BI(): return sys.stdin.buffer.readline().rstrip()
def SI(): return sys.stdin.buffer.readline().rstrip().decode()
dij = [(0, 1), (-1, 0), (0, -1), (1, 0)]
inf = 10**16
# md = 998244353
md = 10**9+7

def nCr(com_n, com_r):
    if com_n < com_r: return 0
    return fac[com_n]*ifac[com_r]%md*ifac[com_n-com_r]%md

# 準備
n_max = 10005
fac = [1]
for i in range(1, n_max+1): fac.append(fac[-1]*i%md)
ifac = [1]*(n_max+1)
ifac[n_max] = pow(fac[n_max], md-2, md)
for i in range(n_max-1, 1, -1): ifac[i] = ifac[i+1]*(i+1)%md
inv = [0]
for i in range(1, n_max): inv.append(pow(i, md-2, md))

n, k = MI()
n += 1
pw = [1]
for _ in range(n_max): pw.append(pw[-1]*n%md)
bb = [1, 500000003]+[0]*(k+1)
for i in range(2, k+3, 2):
    s = 0
    for j in range(i):
        s += nCr(i+1, j)*bb[j]
        s %= md
    bb[i] = s*-inv[i+1]%md
# print(bb)

ans = 0
for i in range(k+1):
    ans += nCr(k+1, i)*bb[i]*pw[k-i+1]
    ans %= md
ans = ans*inv[k+1]%md
print(ans)
0