結果
問題 | No.117 組み合わせの数 |
ユーザー | shobonvip |
提出日時 | 2022-03-11 21:00:23 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 584 bytes |
コンパイル時間 | 225 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 155,008 KB |
最終ジャッジ日時 | 2024-09-16 00:52:26 |
合計ジャッジ時間 | 1,858 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ソースコード
mod = 10**9 + 7 N = 5 * 10**6 fact = [1]*(N+1) factinv = [1]*(N+1) for i in range(2, N+1): fact[i] = fact[i-1] * i % mod factinv[-1] = pow(fact[-1], mod-2, mod) for i in range(N-1, 1, -1): factinv[i] = factinv[i+1] * (i+1) % mod def cmb(a, b): if (a < b) or (b < 0): return 0 return fact[a] * factinv[b] % mod * factinv[a-b] % mod t = int(input()) for i in range(t): s = input() l, r = map(int,s[2:-1].split(",")) if s[0] == "C": print(cmb(l, r)) elif s[0] == "P": if l >= r: print(fact[l] * factinv[l-r] % mod) else: print(0) else: print(cmb(l+r-1, l-1))