結果
問題 | No.1035 Color Box |
ユーザー |
![]() |
提出日時 | 2020-04-24 21:56:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 234 ms / 2,000 ms |
コード長 | 1,297 bytes |
コンパイル時間 | 153 ms |
コンパイル使用メモリ | 82,092 KB |
実行使用メモリ | 159,696 KB |
最終ジャッジ日時 | 2024-10-15 02:46:51 |
合計ジャッジ時間 | 7,200 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
class Combination:"""階乗とその逆元のテーブルをO(N)で事前作成し、組み合わせの計算をO(1)で行う"""def __init__(self, n, MOD):self.fact = [1]for i in range(1, n + 1):self.fact.append(self.fact[-1] * i % MOD)self.inv_fact = [0] * (n + 1)self.inv_fact[n] = pow(self.fact[n], MOD - 2, MOD)for i in reversed(range(n)):self.inv_fact[i] = self.inv_fact[i + 1] * (i + 1) % MODself.MOD = MODdef factorial(self, k):"""k!を求める O(1)"""return self.fact[k]def inverse_factorial(self, k):"""k!の逆元を求める O(1)"""return self.inv_fact[k]def permutation(self, k, r):"""kPrを求める O(1)"""if k < r:return 0return (self.fact[k] * self.inv_fact[k - r]) % self.MODdef combination(self, k, r):"""kCrを求める O(1)"""if k < r:return 0return (self.fact[k] * self.inv_fact[k - r] * self.inv_fact[r]) % self.MODn, m = map(int, input().split())MOD = 10 **9 + 7comb = Combination(10 ** 6, MOD)k = mans = 0for i in range(k + 1):ans += pow(-1, i, MOD) * comb.combination(k, i) * pow(k - i, n, MOD)ans %= MODprint(ans % MOD)