結果

問題 No.117 組み合わせの数
ユーザー fmhrfmhr
提出日時 2015-10-03 01:56:13
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 692 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 82,140 KB
実行使用メモリ 123,372 KB
最終ジャッジ日時 2024-07-19 19:11:01
合計ジャッジ時間 1,432 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

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

ソースコード

diff #

# coding:utf-8

MOD = 1000000007
fact = [1]*2000010
inv = [1]*2000010
invfact = [1]*2000010
for i in range(2, 2000010, 1):
    inv[i] = inv[MOD%i]*(MOD-MOD//i)%MOD
    fact[i] = fact[i-1]*i%MOD
    invfact[i] = invfact[i-1]*inv[i]%MOD


def c(n, k):
    if k<0 or k>n:return 0
    return fact[n]*invfact[k]*invfact[n-k]%MOD


def p(n, k):
    if k<0 or k>n:return 0
    return fact[n]*fact[n-k]%MOD


def h(n, k):
    if n==0 and k==0:return 1
    if n<=0 or k<0:return 0
    return c(n+k-1, k)


T = int(input())
for i in range(T):
    s = input()
    x = s[0]
    n, k = map(int, s[2:-1].split(','))
    if x=="C": print(c(n, k))
    if x=="P": print(p(n, k))
    if x=="H": print(h(n, k))
0