結果
問題 | No.1035 Color Box |
ユーザー |
|
提出日時 | 2021-09-10 10:04:39 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 437 ms / 2,000 ms |
コード長 | 647 bytes |
コンパイル時間 | 315 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 18,432 KB |
最終ジャッジ日時 | 2025-01-02 16:59:38 |
合計ジャッジ時間 | 8,646 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
N, M = map(int, input().split())MOD = 10 ** 9 + 7MAX = 10 ** 5 + 10# https://tjkendev.github.io/procon-library/python/math/factorial.html から拝借しています。fact = [1] * (MAX + 1)rfact = [1] * (MAX + 1)r = 1for i in range(1, MAX + 1):fact[i] = r = r * i % MODrfact[MAX] = r = pow(fact[MAX], MOD - 2, MOD)for i in range(MAX, 0, -1):rfact[i - 1] = r = r * i % MODdef comb(n, k, error=0):if not 0 <= k <= n:return errorreturn fact[n] * rfact[k] * rfact[n - k] % MODans = 0sign = 1for i in range(M, 0, -1):ans += sign * pow(i, N, MOD) * comb(M, i)ans %= MODsign *= -1print(ans)