結果
問題 | No.391 CODING WAR |
ユーザー |
![]() |
提出日時 | 2021-01-25 22:36:33 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 163 ms / 2,000 ms |
コード長 | 839 bytes |
コンパイル時間 | 140 ms |
コンパイル使用メモリ | 82,124 KB |
実行使用メモリ | 88,656 KB |
最終ジャッジ日時 | 2024-06-22 19:08:20 |
合計ジャッジ時間 | 2,447 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 16 |
ソースコード
# coding:UTF-8 import sys MOD = 10 ** 9 + 7 INF = 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 0 elif n < 0 or k < 0: return 0 else: return f[0][n] * (f[1][k] * f[1][n - k] % MOD) % MOD N, M = list(map(int, input().split())) # スペース区切り連続数字 if M > N: print(0) exit() f = combInit(M) res = 0 for i in range(M): if i % 2 == 0: res = (res + comb(M, i, f) * pow(M-i, N, MOD)) % MOD else: res = (res - comb(M, i, f) * pow(M-i, N, MOD)) % MOD print("{}".format(res))