結果

問題 No.3394 Big Binom
コンテスト
ユーザー Ponzu
提出日時 2025-12-01 21:08:57
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
RE  
実行時間 -
コード長 1,159 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,131 ms
コンパイル使用メモリ 278,552 KB
実行使用メモリ 620,644 KB
最終ジャッジ日時 2025-12-01 21:09:04
合計ジャッジ時間 7,315 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 RE * 2
other MLE * 1 -- * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:6:9: warning: #pragma once in main file
    6 | #pragma once
      |         ^~~~

ソースコード

diff #
raw source code 

// ,,廿_廿,,
#include<iostream>
#include<vector>
using namespace std;

#pragma once

#ifndef call_include
#define call_include
#include <bits/stdc++.h>
using namespace std;
#endif

struct NCR {
private:
	vector<long long> comb, inv, finv;
	long long P;

	void calc_inv(long long r) {
		inv[1] = finv[0] = finv[1] = 1LL;
		for(int i = 2; i <= r; i++) {
			inv[i] = P - inv[P % i] * (P / i) % P;
			finv[i] = finv[i - 1] * inv[i] % P;
		}
	}

public:
	NCR(long long n, long long r, long long p) : comb(n + 1), inv(r + 1), finv(r + 1), P(p) {
		if(n / 2 < r) r = n / 2;
		calc_inv(r);

		comb[0] = 1;
		for(int i = 1; i <= r; i++) {
			comb[i] = comb[i - 1] * (n - i + 1) % P * inv[i] % P;
		}
	}

	NCR(long long r, long long p) : inv(r + 1), finv(r + 1), P(p) {
		calc_inv(r);
	}

	long long calc(long long n, long long r) {
		assert(r >= 0);
		if(r > n) return 0;
		if(r > n / 2) r = n - r;
		if(comb.size() > 0) return comb[r];
		else {
			long long f = 1;
			for(long long i = n; i > n - r; i--) (f *= i) %= P;
			return f * finv[r] % P;
		}
	}
};


int main(){
    long long n,k;cin>>n>>k;
    NCR ncr(n,k,998244353);
    cout << ncr.calc(n,k) << endl;
}
0