結果

問題 No.3394 Big Binom
コンテスト
ユーザー V_Melville
提出日時 2025-12-01 15:13:27
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
TLE  
実行時間 -
コード長 1,190 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 995 ms
コンパイル使用メモリ 93,480 KB
実行使用メモリ 7,852 KB
最終ジャッジ日時 2025-12-01 15:13:42
合計ジャッジ時間 14,350 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 modPow(long long base, long long exp, long long mod) {
    long long result = 1;
    base %= mod;
    while (exp > 0) {
        if (exp & 1) 
            result = (result * base) % mod;
        base = (base * base) % mod;
        exp >>= 1;
    }
    return result;
}

long long smallC(long long n, long long k) {
    if (k < 0 || k > n) return 0;
    if (k == 0 || k == n) return 1;
    k = min(k, n - k);
    
    long long num = 1, den = 1;
    for (long long i = 0; i < k; ++i) {
        num = num * ((n - i) % MOD) % MOD;
        den = den * (i + 1) % MOD;
    }
    return num * modPow(den, MOD - 2, MOD) % MOD;
}

long long lucas(long long n, long long k) {
    if (k < 0 || k > n) return 0;
    long long res = 1;
    while (n || k) {
        long long ni = n % MOD;
        long long ki = k % MOD;
        if (ki > ni) return 0;
        res = res * smallC(ni, ki) % MOD;
        n /= MOD;
        k /= MOD;
    }
    return res;
}

int main() {
    long long N, K;
    cin >> N >> K;
    
    if (K > N-K) K = N-K; 
    
    cout << lucas(N, K) << '\n';
    
    return 0;
}
0