結果
問題 | No.117 組み合わせの数 |
ユーザー |
![]() |
提出日時 | 2020-02-28 23:54:17 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 470 ms / 5,000 ms |
コード長 | 1,578 bytes |
コンパイル時間 | 170 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 255,744 KB |
最終ジャッジ日時 | 2024-10-13 19:26:10 |
合計ジャッジ時間 | 1,506 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 1 |
ソースコード
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.MODimport reimport sysinput = sys.stdin.readlinet = int(input())query = [re.split('[(,)]', input()) for i in range(t)]MOD = 10 ** 9 + 7comb = Combination(2 * 10 ** 6 + 10, MOD)for i in range(t):char, n, r, _ = query[i]n = int(n)r = int(r)if char == "P":print(comb.permutation(n, r))elif char == "C":print(comb.combination(n, r))else:if n == r and n == 0:print(1)else:print(comb.combination(n + r - 1, r))