結果

問題 No.3394 Big Binom
コンテスト
ユーザー V_Melville
提出日時 2025-12-01 21:57:26
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
TLE  
実行時間 -
コード長 1,185 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 790 ms
コンパイル使用メモリ 95,464 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-12-01 21:57:39
合計ジャッジ時間 13,138 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 19 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <algorithm>

using namespace std;

const long long MOD = 998244353;

long long power(long long a, long long b, long long m) {
    long long res = 1;
    a %= m;
    while (b > 0) {
        if (b & 1) {
            res = (res * a) % m;
        }
        a = (a * a) % m;
        b >>= 1; 
    }
    return res;
}

long long modInverse(long long a, long long m) {
    return power(a, m - 2, m);
}

long long solve_nck(long long N, long long K) {
    if (K == 0 || K == N) {
        return 1;
    }
    if (K < 0 || K > N) {
        return 0;
    }
    
    if (N%MOD == 0) return 0;

    if (K > N - K) {
        K = N - K;
    }

    long long numerator = 1;   
    long long denominator = 1; 

    for (long long i = 1; i <= K; ++i) {
        numerator = (numerator * (N - i + 1)) % MOD;

        denominator = (denominator * i) % MOD;
    }

    long long inv_denominator = modInverse(denominator, MOD);
    
    return (numerator * inv_denominator) % MOD;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    long long N, K;
    cin >> N >> K;

    long long result = solve_nck(N, K);
    cout << result << endl;

    return 0;
}
0