結果

問題 No.117 組み合わせの数
ユーザー Ratstar216Ratstar216
提出日時 2023-09-12 19:28:45
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 313 ms / 5,000 ms
コード長 1,176 bytes
コンパイル時間 702 ms
コンパイル使用メモリ 69,648 KB
実行使用メモリ 34,688 KB
最終ジャッジ日時 2024-06-30 07:08:19
合計ジャッジ時間 1,826 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include <iostream>
using namespace std;

long long MOD = 1000000007;
long long const sz = 2 * 1000000;
long long fac[sz], facinv[sz];
long long inverse(long long x){
	long long ret = 1;
	long long k = MOD - 2;
	while (k > 0){
		if (k & 1) ret = ret * x % MOD;
		x = x * x % MOD;
		k /= 2;
	}
	return ret;
}
void init(){
	fac[0] = 1;
	for (int i = 1; i < sz; i++) fac[i] = fac[i - 1] * i % MOD;
	facinv[sz - 1] = inverse(fac[sz - 1]);
	for (int i = sz - 2; i > -1; i--) facinv[i] = facinv[i + 1]  * (i + 1) % MOD;
}
long long com(long long n, long long r){
    if (n < r) return 0;
	return fac[n] * facinv[n - r] % MOD * facinv[r] % MOD;
}
long long per(long long n, long long r){
    if (n < r) return 0;
	return fac[n] * facinv[n - r] % MOD;
}
long long hom(long long n, long long r){
	if (n == 0 && r == 0){
		return 1;
	}
	return com(n + r - 1, r);
}
int main(){
    init();
    int t;
    cin >> t;
    for (int _ = 0; _ < t; _++){
        char a;
        char c;
        int n,r;
        cin >> c >> a >> n >> a >> r >> a;
        if (c == 'C') cout << com(n,r) << endl;
        else if (c == 'P') cout << per(n,r) << endl;
        else cout << hom(n,r) << endl;
    }
}
0