結果

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

ソースコード

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