結果
問題 | No.1035 Color Box |
ユーザー |
![]() |
提出日時 | 2021-03-20 11:00:26 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 364 ms / 2,000 ms |
コード長 | 1,359 bytes |
コンパイル時間 | 93 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 18,816 KB |
最終ジャッジ日時 | 2024-11-20 20:04:24 |
合計ジャッジ時間 | 6,028 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
mod = 10 ** 9 + 7class Combination:"""SIZEが10**6程度以下の二項係数を何回も呼び出したいときに使う使い方:comb = Combination(SIZE, MOD)comb(10, 3) => 120"""def __init__(self, N, MOD=10 ** 9 + 7):self.MOD = MODself.__make_factorial_list(N)def __call__(self, n, k):if k < 0 or k > n:return 0res = self.fact[n] * self.inv[k] % self.MODres = res * self.inv[n - k] % self.MODreturn resdef nPk(self, n, k):if k < 0 or k > n:return 0return self.fact[n] * self.inv[n - k] % self.MODdef nHk(self, n, k):if k == 0:return 1return self.__call__(n + k - 1, k)def __make_factorial_list(self, N):self.fact = [1] * (N + 1)self.inv = [1] * (N + 1)MOD = self.MODfor i in range(1, N + 1):self.fact[i] = (self.fact[i - 1] * i) % MODself.inv[N] = pow(self.fact[N], MOD - 2, MOD)for i in range(N, 0, -1):self.inv[i - 1] = (self.inv[i] * i) % MODreturnN, M = map(int, input().split())comb = Combination(10**5+1000, mod)ans = 0f = 1for x in reversed(range(1, M + 1)):cnt = comb(M, x) * pow(x, N, mod) % modans += f * cntans %= modf = -fprint(ans)