結果

問題 No.117 組み合わせの数
ユーザー ooaiuooaiu
提出日時 2024-12-15 23:27:10
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 62 ms / 5,000 ms
コード長 1,911 bytes
コンパイル時間 3,681 ms
コンパイル使用メモリ 251,960 KB
実行使用メモリ 20,596 KB
最終ジャッジ日時 2024-12-15 23:27:15
合計ジャッジ時間 4,703 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#ifndef LOCAL
#include <bits/stdc++.h>

using namespace std;

#define debug(...) (void(0))
#else
#include "algo/debug.h"
#endif

#include <atcoder/modint>
using mint = atcoder::modint1000000007;

template <class Tp>
struct Binom {
   private:
    std::vector<Tp> fact, ifac;
    void make_table(size_t N) {
        const size_t a = fact.size();
        if(a > N) return;
        fact.resize(N + 1);
        for(size_t i = a; i <= N; i++) fact[i] = fact[i - 1] * i;
        ifac.resize(N + 1);
        ifac[N] = Tp{1} / fact[N];
        for(size_t i = N; i >= a; i--) ifac[i - 1] = ifac[i] * i;
    }

   public:
    Binom(size_t N = 0) : fact(1, 1), ifac(1, 1) {
        make_table(N);
    }
    Tp factrial(size_t x) {
        make_table(x);
        return fact[x];
    }
    Tp invfact(size_t x) {
        make_table(x);
        return ifac[x];
    }
    Tp perm(size_t N, size_t K) {
        if(N < K) return Tp{0};
        make_table(N);
        return fact[N] * ifac[N - K];
    }
    Tp comb(size_t N, size_t K) {
        if(N < K) return Tp{0};
        make_table(N);
        return fact[N] * ifac[K] * ifac[N - K];
    }
    Tp homo(size_t N, size_t K) {
        if(N == 0) return K == 0 ? Tp{1} : Tp{0};
        return comb(N - 1 + K, K);
    }
};

Binom<mint> C;

void solve() {
    char c; cin >> c;
    string s; cin >> s;
    int id = 1;
    const auto ri = [&]() -> int {
        int res{};
        for(; s[id] != ',' && s[id] != ')'; id++) {
            res *= 10;
            res += s[id] - '0';
        }
        return res;
    };
    int N = ri();
    id++;
    int K = ri();
    if(c == 'C') cout << C.comb(N, K).val() << '\n';
    else if(c == 'P') cout << C.perm(N, K).val() << '\n';
    else cout << C.homo(N, K).val() << '\n';
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int tt = 1;
    std::cin >> tt;
    while (tt--) {
        solve();
    }
}
0