結果
| 問題 |
No.1035 Color Box
|
| コンテスト | |
| ユーザー |
tpyneriver
|
| 提出日時 | 2020-04-24 21:43:16 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 202 ms / 2,000 ms |
| コード長 | 624 bytes |
| コンパイル時間 | 145 ms |
| コンパイル使用メモリ | 82,028 KB |
| 実行使用メモリ | 121,124 KB |
| 最終ジャッジ日時 | 2024-10-15 02:38:22 |
| 合計ジャッジ時間 | 6,354 ms |
|
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 36 |
ソースコード
import sys
readline = sys.stdin.readline
MOD = 10**9+7
def frac(limit):
frac = [1]*limit
for i in range(2,limit):
frac[i] = i * frac[i-1]%MOD
fraci = [None]*limit
fraci[-1] = pow(frac[-1], MOD -2, MOD)
for i in range(-2, -limit-1, -1):
fraci[i] = fraci[i+1] * (limit + i + 1) % MOD
return frac, fraci
def comb(a, b):
if not a >= b >= 0:
return 0
return frac[a]*fraci[b]*fraci[a-b]%MOD
frac, fraci = frac(1341398)
N, M = map(int, readline().split())
ans = pow(M, N, MOD)
for k in range(1, M):
ans = (ans + (-1)**k * comb(M, k) * pow(M-k, N, MOD))
print(ans % MOD)
tpyneriver