結果
問題 | No.2141 Enumeratest |
ユーザー | kanra824 |
提出日時 | 2022-12-02 22:32:07 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 77 ms / 2,000 ms |
コード長 | 5,808 bytes |
コンパイル時間 | 1,834 ms |
コンパイル使用メモリ | 171,832 KB |
実行使用メモリ | 14,976 KB |
最終ジャッジ日時 | 2024-10-10 00:07:30 |
合計ジャッジ時間 | 3,170 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 9 ms
5,896 KB |
testcase_03 | AC | 2 ms
5,248 KB |
testcase_04 | AC | 77 ms
14,976 KB |
testcase_05 | AC | 2 ms
5,248 KB |
testcase_06 | AC | 2 ms
5,248 KB |
testcase_07 | AC | 5 ms
5,248 KB |
testcase_08 | AC | 10 ms
7,296 KB |
testcase_09 | AC | 7 ms
5,888 KB |
testcase_10 | AC | 13 ms
9,756 KB |
testcase_11 | AC | 21 ms
13,708 KB |
testcase_12 | AC | 2 ms
5,248 KB |
testcase_13 | AC | 2 ms
5,248 KB |
testcase_14 | AC | 20 ms
13,720 KB |
testcase_15 | AC | 13 ms
9,344 KB |
testcase_16 | AC | 17 ms
10,880 KB |
testcase_17 | AC | 5 ms
5,248 KB |
testcase_18 | AC | 3 ms
5,248 KB |
testcase_19 | AC | 6 ms
5,248 KB |
testcase_20 | AC | 4 ms
5,248 KB |
testcase_21 | AC | 4 ms
5,248 KB |
testcase_22 | AC | 3 ms
5,248 KB |
testcase_23 | AC | 3 ms
5,248 KB |
testcase_24 | AC | 32 ms
14,336 KB |
testcase_25 | AC | 10 ms
7,040 KB |
testcase_26 | AC | 2 ms
5,248 KB |
testcase_27 | AC | 4 ms
5,248 KB |
testcase_28 | AC | 3 ms
5,248 KB |
testcase_29 | AC | 3 ms
5,248 KB |
testcase_30 | AC | 5 ms
5,248 KB |
testcase_31 | AC | 5 ms
5,248 KB |
testcase_32 | AC | 6 ms
5,248 KB |
testcase_33 | AC | 4 ms
5,248 KB |
testcase_34 | AC | 47 ms
12,304 KB |
testcase_35 | AC | 2 ms
5,248 KB |
testcase_36 | AC | 4 ms
5,248 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; #define REP(i, n) for (int i = 0; i < (n); ++i) #define RREP(i, n) for (int i = (n); i >= 0; --i) #define FOR(i, a, n) for (int i = (a); i < (n); ++i) #define RFOR(i, a, b) for (int i = (a); i >= (b); --i) #define SZ(x) ((int)(x).size()) #define ALL(x) (x).begin(), (x).end() template <class T> ostream& operator<<(ostream& os, const vector<T>& v) { REP(i, SZ(v)) { if (i) os << " "; os << v[i]; } return os; } template <class T, class U> ostream& operator<<(ostream& os, const pair<T, U>& p) { os << p.first << " " << p.second; return os; } template <class T> bool chmax(T& a, const T& b) { bool res = a < b; if (a < b) a = b; return res; } template <class T> bool chmin(T& a, const T& b) { bool res = a > b; if (a > b) a = b; return res; } using ll = long long; using ull = unsigned long long; using ld = long double; using P = pair<int, int>; using PLL = pair<ll, ll>; using vi = vector<int>; using vll = vector<ll>; using vvi = vector<vi>; using vvll = vector<vll>; const ll MOD = 1e9 + 7; const ll MOD998 = 998244353; const int INF = INT_MAX; const ll LINF = LLONG_MAX; const int inf = INT_MIN; const ll linf = LLONG_MIN; const ld eps = 1e-9; template<int m> struct mint { int x; mint(ll x = 0) : x(((x % m) + m) % m) {} mint operator-() const { return x ? m-x : 0; } mint &operator+=(mint r) { if ((x += r.x) >= m) x -= m; return *this; } mint &operator-=(mint r) { if ((x -= r.x) < 0) x += m; return *this; } mint &operator*=(mint r) { x = ((ll)x * r.x) % m; return *this; } mint inv() const { return pow(m-2); } mint &operator/=(mint r) { return *this *= r.inv(); } friend mint operator+(mint l, mint r) { return l += r; } friend mint operator-(mint l, mint r) { return l -= r; } friend mint operator*(mint l, mint r) { return l *= r; } friend mint operator/(mint l, mint r) { return l /= r; } mint pow(ll n) const { mint ret = 1, tmp = *this; while (n) { if (n & 1) ret *= tmp; tmp *= tmp, n >>= 1; } return ret; } friend bool operator==(mint l, mint r) { return l.x == r.x; } friend bool operator!=(mint l, mint r) { return l.x != r.x; } friend ostream &operator<<(ostream &os, mint a) { return os << a.x; } friend istream &operator>>(istream &is, mint& a) { ll x; is >> x; a = x; return is; } }; using Int = mint<MOD998>; template<typename T> struct Combination { int _n = 1; vector<T> _fact{1}, _rfact{1}; void extend(int n) { if (n <= _n) return; _fact.resize(n); _rfact.resize(n); for (int i = _n; i < n; ++i) _fact[i] = _fact[i - 1] * i; _rfact[n - 1] = 1 / _fact[n - 1]; for (int i = n - 1; i > _n; --i) _rfact[i - 1] = _rfact[i] * i; _n = n; } T fact(int k) { extend(k + 1); return _fact.at(k); } T rfact(int k) { extend(k + 1); return _rfact.at(k); } T P(int n, int r) { if (r < 0 or n < r) return 0; return fact(n) * rfact(n - r); } T C(int n, int r) { if (r < 0 or n < r) return 0; return fact(n) * rfact(r) * rfact(n - r); } T H(int n, int r) { return (n == 0 and r == 0) ? 1 : C(n + r - 1, r); } // O(k logn) // スターリング数 // Stirling(n, k) := n 個の区別できるボールを k 個の区別できない箱にいれる場合の数 // それぞれの箱には1個以上ボールをいれる // --- // S(n, k) = S(n-1, k-1) + k * S(n-1, k) // * 特定の1個だけで1個の箱にいれる場合は S(n-1, k-1) // * そうでない場合は S(n-1, k) 通りに対して特定の1個をいれるのが k 通り // --- // 各グループにつきr個以上, の制限がある場合 // S(n, k) = C(n-1, r-1) * S(n-r, k-1) + k * S(n-1, k) // --- // 玉がn個あるうちのいくつかを選んでkグループに分ける場合 // S(n, k) = S(n-1, k-1) + (k+1) * S(n-1, k) T Stirling(ll n, int k) { T ret = 0; for (int l = 0; l <= k; ++l) { ret += C(k, l) * T{k-l}.pow(n) * (l & 1 ? -1 : 1); } return ret / fact(k); } // O(k^2 logn) // ベル数 // Bell(n, k) := n 個の区別できるボールを k 個の区別できない箱にいれる場合の数 // --- // B(n+1) := Bell(n+1, n+1) = n 個の区別できるボールの分割の総数 // B(n+1) = \sum_{i=0}^n C(n,i) * B(i) // * 特定の1個が属するグループに, 他のボールがn-i 個入っているとき, // * 残りi 個の並べ方はB(i) T Bell(ll n, int k) { T ret = 0; for (int l = 0; l <= k; ++l) { ret += Stirling(n, l); } return ret; } }; int main() { cin.tie(0); ios::sync_with_stdio(false); cout << fixed << setprecision(10); int n, m; cin >> n >> m; Combination<Int> comb; Int ans = 1; if(n > m) { REP(i, m) { ans *= i + 1; } } else { int val = m / n; int cnt = m % n; int now = m; REP(i, n-1) { if(i < cnt) { ans *= comb.C(now, m / n + 1); now -= m / n + 1; } else { ans *= comb.C(now, m / n); now -= m / n; } } } cout << ans << endl; }