結果

問題 No.117 組み合わせの数
ユーザー Ratstar216
提出日時 2023-08-27 02:52:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 3,009 ms / 5,000 ms
コード長 732 bytes
コンパイル時間 499 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 167,296 KB
最終ジャッジ日時 2024-12-26 09:08:22
合計ジャッジ時間 6,964 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

import re

MOD = 10 ** 9 + 7
sz = 2 * 10 ** 6
def inverse(n):
    return pow(n,MOD-2,MOD)
fac = [0] * (sz)
fac[0] = 1
for i in range(1, sz):
    fac[i] = fac[i-1] * i % MOD
facinv = [0] * (sz)
facinv[sz - 1] = inverse(fac[sz - 1])
for i in range(sz - 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