結果
問題 | No.1035 Color Box |
ユーザー |
![]() |
提出日時 | 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 sysreadline = sys.stdin.readlineMOD = 10**9+7def frac(limit):frac = [1]*limitfor i in range(2,limit):frac[i] = i * frac[i-1]%MODfraci = [None]*limitfraci[-1] = pow(frac[-1], MOD -2, MOD)for i in range(-2, -limit-1, -1):fraci[i] = fraci[i+1] * (limit + i + 1) % MODreturn frac, fracidef comb(a, b):if not a >= b >= 0:return 0return frac[a]*fraci[b]*fraci[a-b]%MODfrac, 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)