結果

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

テストケース

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

ソースコード

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) if n * n + r * r > 0 else 1)
0