結果
問題 | No.117 組み合わせの数 |
ユーザー |
|
提出日時 | 2022-12-08 19:30:22 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 560 ms / 5,000 ms |
コード長 | 914 bytes |
コンパイル時間 | 192 ms |
コンパイル使用メモリ | 82,364 KB |
実行使用メモリ | 124,160 KB |
最終ジャッジ日時 | 2024-10-14 16:16:34 |
合計ジャッジ時間 | 1,553 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 1 |
ソースコード
MOD = 10 ** 9 + 7m = 2 * (10 ** 6)fact = [1 for _ in range(m + 1)]inv = [1 for _ in range(m + 1)]ifact = [1 for _ in range(m + 1)]for x in range(2, m + 1):inv[x] = (- (MOD // x) * inv[MOD % x]) % MODfact[x] = (fact[x - 1] * x) % MODifact[x] = (ifact[x - 1] * inv[x]) % MODdef choose(n, k):if n < k:return 0else:return fact[n] * ifact[n - k] * ifact[k] % MODdef permutation(n, k):if n < k:return 0else:return fact[n] * ifact[n - k] % MODdef repeated_permutation(n, k):if n == 0:if k == 0:return 1else:return 0else:return fact[n + k - 1] * ifact[n - 1] * ifact[k] % MODt = int(input())for _ in range(t):ty, *args = list(input())n, k = map(int, ''.join(args[1:-1]).split(','))if ty == 'C':print(choose(n, k))elif ty == 'P':print(permutation(n, k))elif ty == 'H':print(repeated_permutation(n, k))