結果

問題 No.117 組み合わせの数
ユーザー tnodinotnodino
提出日時 2022-06-11 00:10:28
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 4,250 ms / 5,000 ms
コード長 1,283 bytes
コンパイル時間 452 ms
コンパイル使用メモリ 81,780 KB
実行使用メモリ 144,996 KB
最終ジャッジ日時 2023-10-21 06:09:21
合計ジャッジ時間 9,102 ms
ジャッジサーバーID
（参考情報） 		judge9 / judge10
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

mod = 10**9+7
class NumberOfCases:
    def __init__(self, N, mod):
        self.mod = mod
        self.fact = [0] * (N+1)
        self.factinv = [0] * (N+1)
        self.fact[0] = 1
        self.factinv[0] = 1
        for i in range(1,N+1):
            self.fact[i] = (i * self.fact[i-1]) % self.mod
            self.factinv[i] = self.bin_power(self.fact[i], self.mod - 2)

    def bin_power(self, x, k):
        ret = 1
        while k > 0:
            if k & 1:
                ret = ret * x % self.mod
            x = x * x % self.mod
            k >>= 1
        return ret

    def comb(self, n, r):
        if n < r or r < 0:
            return 0
        return self.fact[n] * self.factinv[r] % self.mod * self.factinv[n-r] % self.mod

    def perm(self, n, r):
        if n < r or r < 0:
            return 0
        return self.fact[n] * self.factinv[n-r] % self.mod

    def comb_h(self, n, r):
        if n == r == 0:
            return 1
        return self.comb(n+r-1, r)

def solve():
    S = input()
    x = eval(S[1:])
    if S[0] == 'C':
        print(C.comb(x[0], x[1]))
    if S[0] == 'P':
        print(C.perm(x[0], x[1]))
    if S[0] == 'H':
        print(C.comb_h(x[0], x[1]))

M = 10**6
C = NumberOfCases(M*2, mod)
T = int(input())
for _ in range(T):
    solve()
0