結果
問題 | No.117 組み合わせの数 |
ユーザー | mkawa2 |
提出日時 | 2020-01-20 22:15:42 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 874 bytes |
コンパイル時間 | 278 ms |
コンパイル使用メモリ | 11,024 KB |
実行使用メモリ | 165,232 KB |
最終ジャッジ日時 | 2023-09-15 21:46:53 |
合計ジャッジ時間 | 4,676 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge11 |
(要ログイン)
ソースコード
import sys sys.setrecursionlimit(10 ** 6) def II(): return int(sys.stdin.readline()) def nCr(com_n, com_r): if com_n < com_r: return 0 return fac[com_n] * ifac[com_r] * ifac[com_n - com_r] % md def nPr(com_n, com_r): if com_n < com_r: return 0 return fac[com_n] * ifac[com_n - com_r] % md def nHr(com_n, com_r): return nCr(com_n + com_r - 1, com_r) md = 10 ** 9 + 7 n_max = 10 ** 6 * 2 fac = [1] for i in range(1, n_max + 1): fac.append(fac[-1] * i % md) ifac = [1] * (n_max + 1) ifac[n_max] = pow(fac[n_max], md - 2, md) for i in range(n_max - 1, 1, -1): ifac[i] = ifac[i + 1] * (i + 1) % md def main(): t = II() for _ in range(t): s = input() op = s[0] n, r = map(int, s[2:-1].split(",")) if op == "C": print(nCr(n, r)) if op == "P": print(nPr(n, r)) if op == "H": print(nHr(n, r)) main()