結果

問題 No.2381 Gift Exchange Party
ユーザー sgfcsgfc
提出日時 2023-07-14 23:20:29
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 3,735 bytes
コンパイル時間 3,547 ms
コンパイル使用メモリ 233,160 KB
実行使用メモリ 44,160 KB
最終ジャッジ日時 2024-09-16 08:30:08
合計ジャッジ時間 8,915 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 72 ms
44,160 KB
testcase_01 AC 73 ms
38,656 KB
testcase_02 AC 75 ms
38,656 KB
testcase_03 AC 79 ms
38,656 KB
testcase_04 AC 75 ms
38,528 KB
testcase_05 AC 76 ms
38,656 KB
testcase_06 AC 77 ms
38,528 KB
testcase_07 AC 80 ms
38,656 KB
testcase_08 AC 77 ms
38,784 KB
testcase_09 AC 76 ms
38,656 KB
testcase_10 AC 75 ms
38,528 KB
testcase_11 AC 73 ms
38,528 KB
testcase_12 AC 72 ms
38,528 KB
testcase_13 AC 73 ms
38,528 KB
testcase_14 AC 74 ms
38,656 KB
testcase_15 AC 72 ms
38,656 KB
testcase_16 AC 73 ms
38,528 KB
testcase_17 AC 79 ms
38,528 KB
testcase_18 AC 83 ms
38,656 KB
testcase_19 AC 78 ms
38,656 KB
testcase_20 AC 79 ms
38,656 KB
testcase_21 TLE -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#include <unordered_map>
#include <unordered_set>

#include <stdio.h>
#include <math.h>
#include <cassert>

#pragma warning(push)
#pragma warning(disable:26498 26450 26451 26812)
#include "atcoder/all"
#pragma warning(pop)

//#include <boost/multiprecision/cpp_int.hpp>
//using namespace boost::multiprecision;
// std::fixed << std::setprecision(10) <<

using namespace std;
using namespace atcoder;

using ll = long long;
using ull = unsigned long long;
using mint = modint998244353;
using mint2 = modint1000000007;

const double eps = 1e-9;

#define REP(i, n) for (ll i = 0; i < ll(n); ++i)
#define REPI(i, init, n) for (ll i = init; i < ll(n); ++i)
#define REPD(i, init) for (ll i = init; i >=0; --i)
#define REPDI(i, init, n) for (ll i = init; i >=n; --i)
#define REPE(i, c) for (const auto& i : c)

using vl = vector<ll>;
#define VL(a, n) vector<ll> a; a.assign(n, 0);
#define VLI(a, n) vector<ll> a; a.assign(n, 0); for(auto& x : a) cin >> x;
#define VSI(a, n) vector<string> a; a.assign(n, ""); for(auto& x : a) cin >> x;
using vvl = vector<vector<ll>>;
#define VVLI(a, n, m, init) vector<vector<ll>> a(n); for(auto& x : a) x.assign(m, init);
using pl = pair<ll, ll>;

struct uv { ll u; ll v; ll c; };
#define VUVI(a, n) vector<uv> a; a.assign(n, {0, 0, 1}); for(auto& x : a) {cin >> x.u >> x.v; x.u--; x.v--;}
#define VUVCI(a, n) vector<uv> a; a.assign(n, {0, 0, 0}); for(auto& x : a) {cin >> x.u >> x.v >> x.c; x.u--; x.v--;}
vvl to_edge(const ll n, const vector<uv>& v) { vvl ret(n); for (auto& x : v) ret[x.u].push_back(x.v); return ret; }
vvl to_edge_d(const ll n, const vector<uv>& v) { vvl ret(n); for (auto& x : v) { ret[x.u].push_back(x.v); ret[x.v].push_back(x.u); } return ret; }

template <class T = long long> using pql = priority_queue<T>;
template <class T = long long> using pqg = priority_queue<T, vector<T>, greater<T>>;

const char* YES = "Yes";
const char* NO = "No";
void yn(bool f) { std::cout << (f ? YES : NO) << endl; };
template<class T> void ov(const T& v) { for (auto it = v.begin(); it != v.end(); it++) { if (it != v.begin()) cout << " "; cout << *it; } };

const ll llhuge = 1LL << 60; //十分でかいが多少足しても溢れない数値
template<class T, class U> bool chmin(T& a, const U& b) { a = min(a, b); return b == a; }
template<class T, class U> bool chmax(T& a, const U& b) { a = max(a, b); return b == a; }
template<class T> void mysort(T& v) { std::sort(begin(v), end(v)); };
template<class T, class U> void mysort(T& v, U pr) { std::sort(begin(v), end(v), pr); };
template<class T> void myrev(T& v) { std::reverse(begin(v), end(v)); };
using vm = vector<mint>;
using vvm = vector<vm>;

using _mint = atcoder::modint998244353;

_mint fac[3000001], finv[3000001], inv[3000001];

void setup() {
	const int MOD = _mint::mod();
	fac[0] = fac[1] = 1;
	finv[0] = finv[1] = 1;
	inv[1] = 1;
	for (int i = 2; i <= 3000000; i++) {
		fac[i] = fac[i - 1] * i;
		inv[i] = (_mint)MOD - inv[MOD % i] * (MOD / i);
		finv[i] = finv[i - 1] * inv[i];
	}
}

_mint binom(int n, int r) {
	if (n < r or n < 0 or r < 0) return _mint::raw(0);
	return fac[n] * finv[r] * finv[n - r];
}
int main() {
	cin.tie(nullptr);
	ios::sync_with_stdio(false);

	ll n, p;
	cin >> n >> p;

	setup();

	mint all = 1; //全組み合わせ
	REP(i, n) {
		all *= (i + 1);
	}

	mint ngp = 1; //(p-1)!
	REP(i, p-1) {
		ngp *= (i + 1);
	}

	mint ng = 0;
	mint npg2 = 1;
	mint fiv = 1;
	REP(i, n / p + 1) {
		ng += binom(n, p*i) * fac[p*i] * fiv * finv[i] * npg2;
		npg2 *= ngp;
		fiv *= finv[p];
	}

	//n個のものからp*i個を選ぶ  binom(n, p*i);
	//p*i個をp個ずつi種類に分ける (p*i)! / (p!)^i = fac[p*i] * finv[p]^i
	//p個の並び替え = p!

	cout << (all - ng).val() << endl;
}
0