結果

問題 No.117 組み合わせの数
ユーザー glretoglreto
提出日時 2021-03-31 19:27:01
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 262 ms / 5,000 ms
コード長 1,916 bytes
コンパイル時間 1,209 ms
コンパイル使用メモリ 98,932 KB
実行使用メモリ 52,912 KB
最終ジャッジ日時 2024-05-08 22:46:00
合計ジャッジ時間 2,343 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include <iostream>
#include<vector>
#include<string>
#include<algorithm>
#include<iomanip>
#include<map>
#include<complex>
#include<math.h>
#include<functional>
#include<stack>
#include<queue>
#include<set>
#include <cassert>
using namespace std;
#define rep(i, n) for (ll i = 0; i < (ll)(n); i++)
#define req(i,n) for(int i = 1;i <=  n; i++)
#define rrep(i,n) for(ll i = n-1;i >= 0;i--)
#define ALL(obj) begin(obj), end(obj)
#define RALL(a) rbegin(a),rend(a)
typedef long long ll;
typedef long double ld;
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }
const ll INF = 1e18;
const int inf = 0x3fffffff;
const int MAX = 2110000;
const ll MOD = 1000000007;
long long fac[MAX], finv[MAX], inv[MAX];
void COMint() {
    fac[0] = fac[1] = 1; inv[1] = 1;
    finv[0] = finv[1] = 1;
    for (int i = 2; i < MAX; i++) {
        fac[i] = fac[i - 1] * i % MOD;
        inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
        finv[i] = finv[i - 1] * inv[i] % MOD;
    }
}
long long COM(int n, int k) {
    if (n < k || n < 0 || k < 0) return 0;
    return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}
long long PER(int n, int m) {
    if (n < m || n < 0 || m < 0) return 0;
    return fac[n] * finv[n - m] % MOD;
}
int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    COMint();
    int t; cin >> t;
    rep(_, t) {
        string s;cin >> s;
        char c = s[0];int i;
        for (i = 2; i < s.size(); i++) if (s[i] == ',') break;
        int n = stoi(s.substr(2, i - 2));
        int k = stoi(s.substr(i + 1, s.size() - i - 2));
        if (c == 'C') cout << COM(n, k) << endl;
        else if (c == 'P') cout << PER(n, k) << endl;
        else {
            if (n == 0 && k == 0) cout << 1 << endl;
            else cout << COM(n + k - 1, k) << endl;
        }
    }
}
0