結果
問題 | No.117 組み合わせの数 |
ユーザー | unagiunag |
提出日時 | 2020-08-25 00:58:35 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,285 bytes |
コンパイル時間 | 775 ms |
コンパイル使用メモリ | 79,964 KB |
実行使用メモリ | 8,192 KB |
最終ジャッジ日時 | 2024-11-06 10:56:03 |
合計ジャッジ時間 | 1,579 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ソースコード
//#define PROBLEM "https://yukicoder.me/problems/184" // //#include "../../library/library/debug_template.cpp" //#include "../../library/library/Factorial.cpp" #include <cstdio> #include <iostream> #include <iomanip> #include <vector> using namespace std; using lint = long long; constexpr int MOD = 1000000007; class Factorial { private: vector<lint> fac, inv, finv; void build(int N) { fac[0] = fac[1] = 1; inv[1] = 1; finv[0] = finv[1] = 1; for (int i = 2; i < N; i++) { fac[i] = fac[i - 1] * i % MOD; inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD; finv[i] = finv[i - 1] * inv[i] % MOD; } } public: Factorial(int N = 110000) : fac(N + 1), inv(N + 1), finv(N + 1) { build(N + 1); } lint Cmod(int n, int k) { if (n < k || k < 0) return 0LL; return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD; } lint Pmod(int n, int k) { if (n < k || k < 0) return 0LL; return fac[n] * finv[n - k] % MOD; } }; int main() { Factorial F(200000); int t; scanf("%d\n", &t); for (int i = 0; i < t; i++) { char c; int n, k; scanf("%c(%d,%d)\n", &c, &n, &k); if (c == 'C') { printf("%lld\n", F.Cmod(n, k)); } else if (c == 'P') { printf("%lld\n", F.Pmod(n, k)); } else { printf("%lld\n", F.Cmod(n + k - 1, k)); } } return 0; }