結果

問題 No.117 組み合わせの数
ユーザー RyuRyu
提出日時 2018-06-07 17:43:00
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,822 bytes
コンパイル時間 1,639 ms
コンパイル使用メモリ 169,392 KB
実行使用メモリ 34,660 KB
最終ジャッジ日時 2024-06-30 10:32:55
合計ジャッジ時間 3,023 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

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 = 2000010;

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;
    }

    int T;
    cin >> T;

    char com;
    int N, K;

    REP(i, T)
    {
        scanf(" %c(%d,%d)", &com, &N, &K);

        int ans = 0;

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

    return 0;
}
0