結果

問題 No.117 組み合わせの数
ユーザー RyuRyu
提出日時 2018-06-07 17:26:41
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 1,922 bytes
コンパイル時間 1,514 ms
コンパイル使用メモリ 170,376 KB
実行使用メモリ 20,552 KB
最終ジャッジ日時 2023-09-12 23:06:41
合計ジャッジ時間 3,249 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include <bits/stdc++.h>

#define REP(i, n) for (int i = 0; i < (n); i++)
#define FOR(i, init, n) for (int i = init; i < (n); i++)
#define ALL(obj) (obj).begin(), (obj).end()
#define fi first
#define se second

using namespace std;
using ll = long long int;
using P = pair<int, int>;
using T = tuple<int, int, int>;
using edge = struct
{
    int to, cost;
};

const int MOD = 1e9 + 7;
const int iINF = 1e9;
const long long int llINF = 1e18;
const double PI = acos(-1.0);

const int dx[4] = {1, 0, -1, 0};
const int dy[4] = {0, 1, 0, -1};

const int MAX = 1000010;

vector<ll> fac(MAX), ifac(MAX);

ll mpow(ll x, ll n)
{
    ll ans = 1;

    while (n > 0)
    {
        if (n & 1)
            ans = ans * x % MOD;

        x = x * x % MOD;

        n = n >> 1;
    }

    return ans;
}


ll perm(ll n, ll k)
{
    if(n < k) return 0;

    return (fac[n] * ifac[n - k]) % MOD;
}

ll dupc(ll n, ll k)
{
    if (n == 0 && k == 0)
        return 1;

    if (n < 0 || k == 0)
        return 0;

    return ((fac[n + k - 1] * ifac[k]) % MOD * ifac[n - 1]) % MOD;
}

ll comb(ll n, ll k)
{
    if (n == 0 && k == 0)
        return 1;

    if (n < k || n < 0)
        return 0;

    return ((fac[n] * ifac[k]) % MOD * ifac[n - k]) % MOD;
}

int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);

    fac[0] = 1;
    ifac[0] = 1;

    REP(i, MAX)
    {
        fac[i + 1] = fac[i] * (i + 1) % MOD;

        ifac[i + 1] = ifac[i] * mpow(i + 1, MOD - 2) % MOD;
    }

    ll T;
    cin >> T;

    vector<char> coms(T);
    vector<ll> N(T), K(T);
    
    char a, b, c, d, e;
    
    REP(i, T) cin >> coms[i] >> a >> N[i] >> b >> K[i] >> c;

    REP(i, T)
    {
        ll ans = 0;

        if(coms[i] == 'P')
            ans = perm(N[i], K[i]);
        else if(coms[i] == 'C')
            ans = comb(N[i], K[i]);
        else
            ans = dupc(N[i], K[i]);
         
         cout << ans << endl;
    }

    return 0;
}
0