結果

問題 No.2381 Gift Exchange Party
ユーザー cn_449cn_449
提出日時 2023-07-14 21:42:54
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 77 ms / 2,000 ms
コード長 768 bytes
コンパイル時間 4,296 ms
コンパイル使用メモリ 264,224 KB
実行使用メモリ 5,760 KB
最終ジャッジ日時 2024-09-16 06:36:54
合計ジャッジ時間 6,457 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;

vector<mint> f,finv;
void calc(int n){
	f.resize(n);
	finv.resize(n);
	f[0] = 1;
	finv[0] = 1;
	for(int i = 1; i < n; i++){
		f[i] = f[i - 1] * i;
		finv[i] = finv[i - 1] / i;
	}
}

mint comb(int n, int r){
	return f[n] * finv[r] * finv[n - r];
}

int main(){
	calc(300000);
	ll n, p;
	cin >> n >> p;
	mint ans = 0;
	mint pre = 1;
	for(int i = n; i >= 0; i--){
		if(n % p == i % p){
			mint v = comb(n, i);
			int r = n - i;
			if(r) for(int j = 0; j < p; j++) pre *= r - j;
			mint tmp = pre;
			tmp /= mint(p).pow(r / p);
			tmp *= finv[r / p];
			ans += v * tmp;
		}
	}
	ans = f[n] - ans;
	cout << ans.val() << '\n';
}
0