結果
| 問題 |
No.391 CODING WAR
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2021-09-05 23:29:44 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 599 ms / 2,000 ms |
| コード長 | 650 bytes |
| コンパイル時間 | 489 ms |
| コンパイル使用メモリ | 81,712 KB |
| 実行使用メモリ | 246,272 KB |
| 最終ジャッジ日時 | 2024-12-22 19:01:52 |
| 合計ジャッジ時間 | 12,482 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 16 |
ソースコード
n, k = map(int, input().split())
#以下コピペ
def cmb(n, r, p):
if (r < 0) or (n < r):
return 0
r = min(r, n - r)
return fact[n] * factinv[r] * factinv[n-r] % p
p = 10**9+7
N = 2*(10 ** 6) # N は必要分だけ用意する
fact = [1, 1] # fact[n] = (n! mod p)
factinv = [1, 1] # factinv[n] = ((n!)^(-1) mod p)
inv = [0, 1] # factinv 計算用
for i in range(2, N + 1):
fact.append((fact[-1] * i) % p)
inv.append((-inv[p % i] * (p // i)) % p)
factinv.append((factinv[-1] * inv[-1]) % p)
#以上コピペ
ans = 0
for i in range(k+1):
ans += (-1)**(k-i)*cmb(k, i, p)*pow(i, n, p)
ans %= p
print(ans)