#822179 (C++17) No.2141 Enumeratest

提出ソース

結果

問題	No.2141 Enumeratest
ユーザー	Daylight
提出日時	2022-12-03 12:24:33
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	150 ms / 2,000 ms
コード長	3,975 bytes
コンパイル時間	3,965 ms
コンパイル使用メモリ	266,744 KB
実行使用メモリ	15,148 KB
最終ジャッジ日時	2024-10-10 15:06:01
合計ジャッジ時間	6,524 ms
ジャッジサーバーID （参考情報）	judge5 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,820 KB
testcase_01	AC	2 ms 6,820 KB
testcase_02	AC	32 ms 6,816 KB
testcase_03	AC	150 ms 11,284 KB
testcase_04	AC	20 ms 11,152 KB
testcase_05	AC	19 ms 11,036 KB
testcase_06	AC	2 ms 6,820 KB
testcase_07	AC	14 ms 6,816 KB
testcase_08	AC	54 ms 6,816 KB
testcase_09	AC	36 ms 6,820 KB
testcase_10	AC	85 ms 7,832 KB
testcase_11	AC	139 ms 15,148 KB
testcase_12	AC	7 ms 6,816 KB
testcase_13	AC	34 ms 6,820 KB
testcase_14	AC	132 ms 12,308 KB
testcase_15	AC	77 ms 7,548 KB
testcase_16	AC	100 ms 13,340 KB
testcase_17	AC	14 ms 8,212 KB
testcase_18	AC	17 ms 9,560 KB
testcase_19	AC	17 ms 10,108 KB
testcase_20	AC	14 ms 8,696 KB
testcase_21	AC	16 ms 9,396 KB
testcase_22	AC	12 ms 7,592 KB
testcase_23	AC	7 ms 6,820 KB
testcase_24	AC	120 ms 14,164 KB
testcase_25	AC	48 ms 6,816 KB
testcase_26	AC	19 ms 10,372 KB
testcase_27	AC	15 ms 9,236 KB
testcase_28	AC	18 ms 10,668 KB
testcase_29	AC	11 ms 7,296 KB
testcase_30	AC	17 ms 10,284 KB
testcase_31	AC	13 ms 8,064 KB
testcase_32	AC	14 ms 8,576 KB
testcase_33	AC	12 ms 7,936 KB
testcase_34	AC	47 ms 10,488 KB
testcase_35	AC	3 ms 6,820 KB
testcase_36	AC	18 ms 10,436 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>

using namespace std;
#define SZ(x) (int) (x).size()
#define REP(i, n) for(int i = 0; i < (n); i++)
#define FOR(i, a, b) for(auto i = (a); i < (b); i++)
#define For(i, a, b, c) \
	for(auto i = (a); i != (b); i += (c))
#define REPR(i, n) for(auto i = (n) -1; i >= 0; i--)
#define ALL(s) (s).begin(), (s).end()
#define so(V) sort(ALL(V))
#define rev(V) reverse(ALL(V))
#define uni(v) v.erase(unique(ALL(v)), (v).end())
#define eb emplace_back

typedef long long unsigned int llu;
typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<bool> vb;
typedef vector<vi> vvi;
typedef vector<vll> vvll;
typedef pair<int, int> PI;
typedef pair<ll, ll> PL;
const double EPS = 1e-9;
const int MOD = 1e9 + 7;
const int INF = (1 << 30);
const ll LINF = 1e18;
const double math_PI = acos(-1);

template<typename T>
vector<T> make_v(size_t a) {
	return vector<T>(a);
}

template<typename T, typename... Ts>
auto make_v(size_t a, Ts... ts) {
	return vector<decltype(make_v<T>(ts...))>(
		a, make_v<T>(ts...));
}

template<typename T, typename V>
typename enable_if<is_class<T>::value == 0>::type fill_v(
	T& t, const V& v) {
	t = v;
}

template<typename T, typename V>
typename enable_if<is_class<T>::value != 0>::type fill_v(
	T& t, const V& v) {
	for(auto& e: t) fill_v(e, v);
}

template<class T>
bool chmax(T& a, const T& b) {
	if(a < b) {
		a = b;
		return true;
	}
	return false;
}

template<class T>
bool chmin(T& a, const T& b) {
	if(a > b) {
		a = b;
		return true;
	}
	return false;
}

template<typename S, typename T>
istream& operator>>(istream& is, pair<S, T>& p) {
	cin >> p.first >> p.second;
	return is;
}

template<typename T>
istream& operator>>(istream& is, vector<T>& vec) {
	for(T& x: vec) is >> x;
	return is;
}

template<typename T>
string join(string splitter, vector<T>& vec) {
	stringstream ss;
	REP(i, SZ(vec)) {
		if(i != 0) ss << splitter;
		ss << vec[i];
	}
	return ss.str();
}

template<typename T>
ostream& operator<<(ostream& os, vector<T>& vec) {
	os << join(" ", vec);
	return os;
}
template<typename T>
struct Comb {
private:
	vector<T> _fact;
	vector<T> inv;
	int pre_build;
	void expand(int x) {
		if(pre_build == -1) return;
		while(SZ(_fact) <= x) {
			int next = SZ(_fact);
			_fact.push_back(_fact[next - 1] * next);
			inv.push_back(_fact[next].inv());
		}
	}

	T nPr_calc(int n, int r) {
		T ret = 1;
		for(int i = n; i > n - r; i--) ret *= i;
		return ret;
	}
	T nCr_calc(int n, int r) {
		T ret = 1;
		REP(i, r) {
			ret *= n - i;
			ret /= i + 1;
		}
		return ret;
	}

public:
	Comb(int pre_build = 0): pre_build(pre_build) {
		if(pre_build == -1) {
			return;
		}
		_fact = vector<T>(pre_build + 1);
		_fact[0] = 1;
		inv = vector<T>(pre_build + 1);
		REP(i, pre_build) {
			_fact[i + 1] = _fact[i] * (i + 1);
		}
		inv[pre_build] = _fact[pre_build].inv();
		REPR(i, pre_build) {
			inv[i] = inv[i + 1] * (i + 1);
		}
	}
	T nPr(int n, int r) {
		assert(r >= 0);
		assert(n >= r);
		expand(n);
		if(pre_build == -1) return nPr_calc(n, r);
		return _fact[n] * inv[n - r];
	}
	T nCr(int n, int r) {
		assert(r >= 0);
		assert(n >= r);
		expand(n);
		if(pre_build == -1) return nCr_calc(n, r);
		return _fact[n] * inv[r] * inv[n - r];
	}
	T nHr(int n, int r) {
		assert(r >= 0);
		assert(n >= 1);
		expand(n + r - 1);
		return nCr(n + r - 1, r);
	}
	T type_permission(vi V) {
		assert(pre_build != -1);
		int sum = 0;
		for(int i: V) {
			assert(i >= 0);
			sum += i;
		}
		expand(sum);
		T ans = _fact[sum];
		for(int i: V) ans *= inv[i];
		return ans;
	}
	T fact(int n) {
		expand(n);
		return _fact[n];
	}
	T inv_fact(int n) {
		expand(n);
		return inv[n];
	}
};

#include <atcoder/all>
using namespace atcoder;

using mint = modint998244353;
int main() {
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	int N, M;
	cin >> N >> M;
	Comb<mint> comb(N);
	mint ans = comb.fact(M);
	REP(i, N) {
		int cur = (M + i) / N;
		ans *= comb.inv_fact(cur);
	}
	cout << ans.val() << endl;
	return 0;
}

yukicoder

結果

テストケース

ソースコード