結果
問題 | No.1035 Color Box |
ユーザー |
![]() |
提出日時 | 2021-01-26 13:02:20 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 110 ms / 2,000 ms |
コード長 | 803 bytes |
コンパイル時間 | 311 ms |
コンパイル使用メモリ | 82,792 KB |
実行使用メモリ | 88,128 KB |
最終ジャッジ日時 | 2024-06-23 12:49:21 |
合計ジャッジ時間 | 3,768 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
# coding:UTF-8import sysMOD = 10 ** 9 + 7INF = float('inf')def combInit(n):fact = [1, 1]finv = [1, 1]inv = [0, 1]for i in range(2, n + 1):fact.append(fact[i - 1] * i % MOD)inv.append(MOD - inv[MOD % i] * (MOD // i) % MOD)finv.append(finv[i - 1] * inv[i] % MOD)return [fact, finv]def comb(n, k, f):if n < k:return 0elif n < 0 or k < 0:return 0else:return f[0][n] * (f[1][k] * f[1][n - k] % MOD) % MODN, M = list(map(int, input().split())) # スペース区切り連続数字f = combInit(M)res = 0for i in range(M):if i % 2 == 0:res = (res + comb(M, i, f) * pow(M-i, N, MOD)) % MODelse:res = (res - comb(M, i, f) * pow(M-i, N, MOD)) % MODprint("{}".format(res))