結果
問題 | No.2130 分配方法の数え上げ mod 998244353 |
ユーザー | Carpenters-Cat |
提出日時 | 2022-11-25 21:49:09 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 18 ms / 2,000 ms |
コード長 | 7,686 bytes |
コンパイル時間 | 1,755 ms |
コンパイル使用メモリ | 135,172 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-02 04:24:18 |
合計ジャッジ時間 | 2,633 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 2 ms
5,248 KB |
testcase_03 | AC | 2 ms
5,248 KB |
testcase_04 | AC | 2 ms
5,248 KB |
testcase_05 | AC | 1 ms
5,248 KB |
testcase_06 | AC | 2 ms
5,248 KB |
testcase_07 | AC | 2 ms
5,248 KB |
testcase_08 | AC | 2 ms
5,248 KB |
testcase_09 | AC | 2 ms
5,248 KB |
testcase_10 | AC | 2 ms
5,248 KB |
testcase_11 | AC | 2 ms
5,248 KB |
testcase_12 | AC | 2 ms
5,248 KB |
testcase_13 | AC | 2 ms
5,248 KB |
testcase_14 | AC | 2 ms
5,248 KB |
testcase_15 | AC | 2 ms
5,248 KB |
testcase_16 | AC | 2 ms
5,248 KB |
testcase_17 | AC | 3 ms
5,248 KB |
testcase_18 | AC | 3 ms
5,248 KB |
testcase_19 | AC | 3 ms
5,248 KB |
testcase_20 | AC | 2 ms
5,248 KB |
testcase_21 | AC | 4 ms
5,248 KB |
testcase_22 | AC | 2 ms
5,248 KB |
testcase_23 | AC | 4 ms
5,248 KB |
testcase_24 | AC | 2 ms
5,248 KB |
testcase_25 | AC | 4 ms
5,248 KB |
testcase_26 | AC | 2 ms
5,248 KB |
testcase_27 | AC | 18 ms
5,248 KB |
testcase_28 | AC | 2 ms
5,248 KB |
testcase_29 | AC | 18 ms
5,248 KB |
testcase_30 | AC | 2 ms
5,248 KB |
testcase_31 | AC | 2 ms
5,248 KB |
testcase_32 | AC | 2 ms
5,248 KB |
testcase_33 | AC | 2 ms
5,248 KB |
testcase_34 | AC | 2 ms
5,248 KB |
testcase_35 | AC | 2 ms
5,248 KB |
testcase_36 | AC | 2 ms
5,248 KB |
testcase_37 | AC | 2 ms
5,248 KB |
ソースコード
#include <algorithm> #include <array> #include <bitset> #include <cassert> #include <cmath> #include <deque> #include <iomanip> #include <iostream> #include <map> #include <queue> #include <random> #include <set> #include <stack> #include <stdio.h> #include <string> #include <tuple> #include <utility> #include <vector> using namespace std; namespace { long long modinv(const long long& a, const long long& m) { long long s = a; long long t = m; long long xs = 1; long long xt = 0; while (s) { long long q = t / s; t -= q * s; std::swap(s, t); xt -= q * s; std::swap(xs, xt); } assert(t == 1); while (xt < 0) { xt += m; } while (xt >= m) { xt -= m; } return xt; } } // namespace class mint { private: long long val; long long raw; public: static long long mod; static long long mod_inv; mint(long long v = 0) : val(v), raw(v) { assert(mod > 1 && mod < (1ll << 31)); if (val < 0) { val = (-val) % mod; val = mod - val; if (val == mod) { val = 0; } } else if (val >= mod) { val %= mod; } } explicit mint(long long v, long long r) : val(v), raw(r) {} mint operator=(const mint& m) { val = m.val; raw = m.raw; return *this; } long long gval() const { return this->val; } long long graw() const { return this->raw; } mint operator+(const mint& m) const { return mint{(val + m.val >= mod ? (val + m.val - mod) : (val + m.val)), raw + m.raw}; } mint operator-(const mint& m) const { return mint{(val - m.val < 0 ? (val - m.val + mod) : (val - m.val)), raw - m.raw}; } mint operator*(const mint& m) const { // とりあえず愚直計算。 // 暇な時に高速なのにしておく return mint{(val * m.val) % mod, raw * m.raw}; } mint inv() const { long long v2 = modinv(val, mod); return v2; } mint operator/(const mint& m) const { return (*this) * m.inv(); } void operator+=(const mint& m) { *this = (*this) + m; } void operator-=(const mint& m) { *this = (*this) - m; } void operator*=(const mint& m) { *this = (*this) * m; } void operator/=(const mint& m) { *this = (*this) / m; } }; void init(long long m) { mint::mod = m; // mint::mod_inv = (-1ull) / m + 1; } std::ostream& operator<<(std::ostream& stream, const mint& m) { const std::string sv = std::to_string(m.gval()); stream << sv; return stream; } class big_mint { }; using ll = long long; using Vi = vector<int>; using VVi = vector<Vi>; using Vl = vector<ll>; using VVl = vector<Vl>; using Pii = pair<int, int>; using Vp = vector<Pii>; using VVp = vector<Vp>; using Pl = pair<ll, int>; using Vpl = vector<Pl>; using VVpl = vector<Vpl>; using tup = tuple<ll, ll, int>; using Vt = vector<tuple<int, int, int>>; using Pll = pair<ll, ll>; using Vc = vector<char>; using VVc = vector<Vc>; template <class U> using PQmax = priority_queue<U>; template <class U> using PQmin = priority_queue<U, vector<U>, greater<U>>; //再帰で計算するトポロジカルソート void tprsort(int u, const VVi& gr, Vi& tpr, Vi& par); //キューで計算するトポロジカルソート void tprsort(const VVi& gr, Vi& tpr); ll mpow(ll x, ll n, ll m = 1e9 + 7); ll comb(int n, int r, const Vl& kai, const Vl& fkai, ll m = 1e9 + 7); ll gcd(ll a, ll b); int LCA(const VVi& par, const Vi& depth, int a, int b); Vi sccResolve(const VVi& gr); void dijkstra(const VVi& gr, const VVl& cost, Vl& dist, int s); ll mint::mod = 998244353; struct S { mint val; ll siz; }; S comp(S a, S b) { return S{a.val + b.val, a.siz + b.siz}; } using F = mint; S mapp(S a, F b) { return S{a.val + b * a.siz, a.siz}; } F fnc(F a, F b) { return F{a + b}; } S e{0, 1}; F id{0}; int main() { ll N, M; cin >> N >> M; ll ans = 0; ll kj = 1; if (N < M) { cout << 0 << endl; return 0; } ll m = 998244353; ll n = N % m; for (int i = 0; i < M; i++) { ans += kj; ans %= m; kj *= (n - i); kj %= m; kj *= mpow(i + 1, m - 2, m); kj %= m; } ll sm = mpow(2, N, mint::mod); ans = sm - ans + m; ans %= m; cout << ans << endl; } // Library void tprsort(int u, const VVi& gr, Vi& tpr, Vi& par) { // idx[u] = tpr.size(); /*for (int v : gr[u]) { if (par[v] == u || par[v] == -1) { par[v] = u; tprsort(v, gr, tpr, par); } } tpr.push_back(u);*/ stack<int> st; st.push(u); bool vis[2000020]; for (int i = 0; i <= gr.size(); i++) { vis[i] = false; } while (!st.empty()) { int v = st.top(); if (!vis[v]) { vis[v] = true; for (int p : gr[v]) { if (par[p] == -1) { par[p] = v; st.push(p); } } } else { tpr.push_back(v); st.pop(); } } } Vi sccResolve(const VVi& gr) { int N = gr.size(); Vi tpr; Vi par(N, -1); for (int i = 0; i < N; i++) { if (par[i] == -1) { tprsort(i, gr, tpr, par); } } Vi ret(N, -1); int now = 0; for (int i = N - 1; i >= 0; i--) { int u = tpr[i]; if (ret[u] != -1) { continue; } ret[u] = now; stack<int> st; st.push(u); while (!st.empty()) { int v = st.top(); st.pop(); for (int p : gr[v]) { if (ret[p] == -1) { st.push(p); ret[p] = now; } } } now++; } return ret; } ll gcd(ll a, ll b) { while (b) { a %= b; swap(a, b); } return a; } ll mpow(ll x, ll n, ll m) { ll ret = 1; while (n) { if (n % 2) { ret *= x; ret %= m; } x = (x * x) % m; n /= 2; } return ret; } ll comb(int n, int r, const Vl& kai, const Vl& fkai, ll m) { if (n < 0 || r < 0 || n < r) { return 0; } ll ret = kai[n]; ret *= fkai[r]; ret %= m; ret *= fkai[n - r]; ret %= m; return ret; } int LCA(const VVi& par, const Vi& depth, int a, int b) { if (depth[a] < depth[b]) { swap(a, b); } int dis = depth[a] - depth[b]; for (int i = 19; i >= 0; i--) { if ((dis >> i) & 1) { a = par[i][a]; } } if (a == b) { return a; } for (int i = 19; i >= 0; i--) { if (par[i][a] != par[i][b]) { a = par[i][a]; b = par[i][b]; } } return par[0][a]; } void dijkstra(const VVi& gr, const VVl& cost, Vl& dist, int s) { ll INF = (ll)1e18; dist.assign(gr.size(), INF); dist[s] = 0; PQmin<Pl> pque; pque.push(make_pair(0, s)); while (!pque.empty()) { auto [L, u] = pque.top(); pque.pop(); while (L != dist[u] && !pque.empty()) { tie(L, u) = pque.top(); pque.pop(); } for (int i = 0; i < gr[u].size(); i++) { int v = gr[u][i]; ll c = cost[u][i]; if (dist[v] > c + L) { dist[v] = c + L; pque.push(make_pair(dist[v], v)); } } } return; }