結果

問題 No.117 組み合わせの数
ユーザー tottoripapertottoripaper
提出日時 2015-03-28 19:04:28
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 238 ms / 5,000 ms
コード長 1,373 bytes
コンパイル時間 425 ms
コンパイル使用メモリ 54,028 KB
実行使用メモリ 19,196 KB
最終ジャッジ日時 2023-09-11 10:52:13
合計ジャッジ時間 1,295 ms
ジャッジサーバーID
(参考情報)
judge12 / judge13
このコードへのチャレンジ(β)

テストケース

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

ソースコード

diff #

// Wrongri-La Shower

#include <iostream>

typedef long long ll;

const ll MOD = 1000000000ll + 7;
ll fact[2000001];

ll mod_pow(ll a, int n){
    ll res = 1ll;
    while(n > 0){
        if(n & 1){res =res * a % MOD;}
        a = a * a % MOD;
        n >>= 1;
    }
    return res;
}

ll mod_inv(ll n){
    return mod_pow(n, MOD-2);
}

ll nPk(int n, int k){
    if(n < k){return 0ll;}

    return fact[n] * mod_inv(fact[n-k]) % MOD;
}

ll nCk(int n, int k){
    if(n < k){return 0ll;}

    return fact[n] * mod_inv(fact[n-k]) % MOD * mod_inv(fact[k]) % MOD;
}

int main(){
    fact[0] = 1ll;
    for(int i=1;i<=2000000;i++){
        fact[i] = fact[i-1] * i % MOD;
    }

    int T;
    std::cin >> T;

    for(;T--;){
        std::string S;
        std::cin >> S;

        char type = S[0];
        int a = 0, b = 0;
        {
            int i = 2;
            while(std::isdigit(S[i])){
                a = a * 10 + (S[i] - '0');
                ++i;
            }

            ++i;

            while(std::isdigit(S[i])){
                b = b * 10 + (S[i] - '0');
                ++i;
            }
        }

        ll res;
        if(type == 'C'){
            res = nCk(a, b);
        }else if(type == 'P'){
            res = nPk(a, b);
        }else{
            res = a == 0 && b == 0 ? 1 : nCk(a+b-1, b);
        }

        std::cout << res << std::endl;
    }
}
0