結果

問題 No.117 組み合わせの数
ユーザー koyopro
提出日時 2015-10-03 14:06:05
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 393 ms / 5,000 ms
コード長 1,184 bytes
コンパイル時間 1,289 ms
コンパイル使用メモリ 158,828 KB
実行使用メモリ 55,424 KB
最終ジャッジ日時 2024-07-19 19:24:21
合計ジャッジ時間 2,676 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:40:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   40 |         scanf(" %c(%lld,%lld)", &c, &n, &k);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;

#define REP(i, n) for(int i=0; i<(n); i++)
#define FOR(i, a, b) for(int i=(a);i<(b);i++)
#define MAX 2222222

#define int long long
int T;
int mod = (int)1e9+7;
int inv[MAX], fact[MAX], finv[MAX];
int perm(int n, int k, int mod=(int)1e9+7) {
    if (k < 0 || n < k) return 0;
    return fact[n] * finv[n-k] % mod;
}
int comb(int n, int k, int mod=(int)1e9+7) {
    if (k < 0 || n < k) return 0;
    return fact[n] * finv[n-k] % mod * finv[k] % mod;
}
int hcomb(int n, int k, int mode=(int)1e9+7) {
    if (n == 0 && k == 0) return 1;
    return comb(n+k-1, k);
}
int pymod(int i, int j) {
    return (i % j + j) % j;
}
signed main()
{
    inv[0] = inv[1] = fact[0] = finv[0] = 1;
    FOR(i,2,MAX) inv[i] = pymod(-mod/i * inv[mod%i], mod);
    FOR(i,1,MAX) {
        fact[i] = i * fact[i-1] % mod;
        finv[i] = inv[i] * finv[i-1] % mod;
    }

    cin >> T;
    char c;
    int n, k;
    REP(i,T) {
        scanf(" %c(%lld,%lld)", &c, &n, &k);
        int ret = 0;
        if (c == 'P') ret = perm(n, k);
        else if (c == 'C') ret = comb(n, k);
        else ret = hcomb(n, k);
        cout << ret << endl;
    }
    return 0;
}
0