結果

問題 No.117 組み合わせの数
ユーザー neterukunneterukun
提出日時 2020-02-28 23:54:17
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 559 ms / 5,000 ms
コード長 1,578 bytes
コンパイル時間 614 ms
コンパイル使用メモリ 87,132 KB
実行使用メモリ 243,712 KB
最終ジャッジ日時 2023-08-03 22:15:35
合計ジャッジ時間 2,356 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 559 ms
243,712 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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) % MOD
        self.MOD = MOD
    
    def 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 0
        return (self.fact[k] * self.inv_fact[k - r]) % self.MOD

    def combination(self, k, r):
        """kCrを求める O(1)"""
        if k < r:
            return 0
        return (self.fact[k] * self.inv_fact[k - r] * self.inv_fact[r]) % self.MOD


import re
import sys
input = sys.stdin.readline


t = int(input())
query = [re.split('[(,)]', input()) for i in range(t)]
MOD = 10 ** 9 + 7
comb = 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))
0