結果

問題 No.117 組み合わせの数
ユーザー Ratstar216Ratstar216
提出日時 2023-08-27 02:46:20
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 741 bytes
コンパイル時間 281 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 167,472 KB
最終ジャッジ日時 2024-06-07 22:38:48
合計ジャッジ時間 5,501 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

import re

MOD = 10 ** 9 + 7
def inverse(n):
    return pow(n,MOD-2,MOD)
fac = [0] * (2 * 10 ** 6)
fac[0] = 1
for i in range(1, 2 * 10 ** 6):
    fac[i] = fac[i-1] * i % MOD
facinv = [0] * (2 * 10 ** 6)
facinv[2 * 10 ** 6 - 1] = inverse(fac[2 * 10 ** 6 - 1])
for i in range(2 * 10 ** 6 - 2, -1, -1):
    facinv[i] = facinv[i+1] * (i+1) % MOD
def bi(n,r):
    if n < r:
        return 0
    return fac[n] * facinv[r] * facinv[n-r] % MOD
def per(n,r):
    if n < r:
        return 0
    return fac[n] * facinv[n-r] % MOD


t = int(input())
for _ in range(t):
    s = input()
    n,r = (map(int,re.findall(r"\d+", s)))
    if s[0] == "C":
        print(bi(n,r))
    elif s[0] == "P":
        print(per(n,r))
    else:
        print(bi(n+r-1,r))
0