結果

問題 No.117 組み合わせの数
ユーザー mkawa2mkawa2
提出日時 2020-01-20 21:54:40
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
RE  
実行時間 -
コード長 1,664 bytes
コンパイル時間 78 ms
コンパイル使用メモリ 10,964 KB
実行使用メモリ 274,736 KB
最終ジャッジ日時 2023-09-15 21:45:24
合計ジャッジ時間 6,777 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

import sys

sys.setrecursionlimit(10 ** 6)
def II(): return int(sys.stdin.readline())

class mint:
    def __init__(self, x):
        self.__x = x % md

    def __str__(self):
        return str(self.__x)

    def __add__(self, other):
        if isinstance(other, mint): other = other.__x
        return mint(self.__x + other)

    def __sub__(self, other):
        if isinstance(other, mint): other = other.__x
        return mint(self.__x - other)

    def __rsub__(self, other):
        return mint(other - self.__x)

    def __mul__(self, other):
        if isinstance(other, mint): other = other.__x
        return mint(self.__x * other)

    __radd__ = __add__
    __rmul__ = __mul__

    def __truediv__(self, other):
        if isinstance(other, mint): other = other.__x
        return mint(self.__x * pow(other, md - 2, md))

    def __pow__(self, power, modulo=None):
        return mint(pow(self.__x, power, md))

def nCr(com_n, com_r):
    if com_n < com_r: return 0
    return fac[com_n] * ifac[com_r] * ifac[com_n - com_r]

def nPr(com_n, com_r):
    if com_n < com_r: return 0
    return fac[com_n] * ifac[com_n - com_r]

def nHr(com_n, com_r):
    return nCr(com_n + com_r, com_r)

md = 10 ** 9 + 7
n_max = 10 ** 6
fac = [mint(1)]
for i in range(1, n_max + 1): fac.append(fac[-1] * i)
ifac = [1] * (n_max + 1)
ifac[n_max] = fac[n_max] ** (md - 2)
for i in range(n_max - 1, 1, -1): ifac[i] = ifac[i + 1] * (i + 1)

def main():
    t=II()
    for _ in range(t):
        s=input()
        op=s[0]
        n,r=map(int,s[2:-1].split(","))
        if op=="C":print(nCr(n,r))
        if op=="P":print(nPr(n,r))
        if op=="H":print(nHr(n,r))

main()
0