結果

問題 No.2544 Many RMQ Problems
ユーザー kusaf_kusaf_
提出日時 2023-12-08 11:10:01
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 24 ms / 2,000 ms
コード長 3,013 bytes
コンパイル時間 2,930 ms
コンパイル使用メモリ 252,232 KB
実行使用メモリ 15,020 KB
最終ジャッジ日時 2023-12-08 11:10:05
合計ジャッジ時間 4,623 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 17 ms
15,020 KB
testcase_01 AC 17 ms
15,020 KB
testcase_02 AC 18 ms
15,020 KB
testcase_03 AC 23 ms
15,020 KB
testcase_04 AC 22 ms
15,020 KB
testcase_05 AC 23 ms
15,020 KB
testcase_06 AC 24 ms
15,020 KB
testcase_07 AC 20 ms
15,020 KB
testcase_08 AC 18 ms
15,020 KB
testcase_09 AC 18 ms
15,020 KB
testcase_10 AC 17 ms
15,020 KB
testcase_11 AC 18 ms
15,020 KB
testcase_12 AC 17 ms
15,020 KB
testcase_13 AC 17 ms
15,020 KB
testcase_14 AC 18 ms
15,020 KB
testcase_15 AC 17 ms
15,020 KB
testcase_16 AC 18 ms
15,020 KB
testcase_17 AC 18 ms
15,020 KB
testcase_18 AC 17 ms
15,020 KB
testcase_19 AC 24 ms
15,020 KB
testcase_20 AC 24 ms
15,020 KB
testcase_21 AC 24 ms
15,020 KB
testcase_22 AC 24 ms
15,020 KB
testcase_23 AC 24 ms
15,020 KB
testcase_24 AC 17 ms
15,020 KB
testcase_25 AC 18 ms
15,020 KB
testcase_26 AC 23 ms
15,020 KB
testcase_27 AC 17 ms
15,020 KB
testcase_28 AC 23 ms
15,020 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

struct Combinatorics {
 private:
  void extend(int m) {
    int n = fac.size();
    m = min(m, mint::mod() - 1);
    if(n > m) { return; }
    fac.resize(m + 1);
    finv.resize(m + 1);
    inv.resize(m + 1);
    for(int i = n; i <= m; i++) { fac[i] = fac[i - 1] * i; }
    finv[m] = fac[m].inv();
    for(int i = m - 1; i >= n; i--) { finv[i] = finv[i + 1] * (i + 1); }
    for(int i = n; i <= m; i++) { inv[i] = finv[i] * fac[i - 1]; }
  }

 public:
  vector<mint> fac, finv, inv;
  Combinatorics(int n = 0): fac(1, 1), finv(1, 1), inv(1, 1) { extend(n); }
  mint operator()(int n) {
    extend(abs(n));
    return n >= 0 ? fac[n] : finv[-n];
  }
  mint operator[](int n) {
    extend(abs(n));
    return n >= 0 ? inv[n] : -inv[-n];
  }
  mint C(int n, int k) {
    if(n < 0 || k < 0 || n < k) { return 0; }
    extend(n);
    return fac[n] * finv[n - k] * finv[k];
  }
  inline mint operator()(int n, int k) { return C(n, k); }
  mint C_(ll n, ll k) {
    if(n < 0 || k < 0 || n < k) { return 0; }
    k = min(k, n - k);
    extend(k);
    mint r = 1;
    for(ll i = 0; i < k; i++) { r *= n - i; }
    return r * finv[k];
  }
  mint P(int n, int k) {
    if(n < 0 || k < 0 || n < k) { return 0; }
    extend(n);
    return fac[n] * finv[n - k];
  }
  mint P_(ll n, ll k) {
    if(n < 0 || k < 0 || n < k) { return 0; }
    k = min(k, n - k);
    mint r = 1;
    for(ll i = 0; i < k; i++) { r *= n - i; }
    return r;
  }
  mint H(int n, int k) {
    if(n < 0 || k < 0) { return 0; }
    if(n == 0 && k == 0) { return 1; }
    return C(n + k - 1, k);
  }
  mint multi(const vector<ll> &v) {
    int n = 0;
    for(auto &i : v) {
      if(i < 0) { return 0; }
      n += i;
    }
    extend(n);
    mint r = fac[n];
    for(auto &i : v) { r *= finv[i]; }
    return r;
  }
  mint operator()(const vector<ll> &v) { return multi(v); }
  mint Catalan(int n) {
    if(n < 0) { return 0; }
    extend(n * 2);
    return fac[n * 2] * finv[n] * finv[n + 1];
  }
  mint Bell(int n, int k) {
    if(n < 0 || k < 0) { return 0; }
    if(n == 0) { return 1; }
    k = min(n, k);
    extend(k);
    mint r = 0;
    vector<mint> v(k + 1);
    v[0] = 1;
    for(ll i = 1; i <= k; i++) { v[i] = v[i - 1] + (i & 1 ? -finv[i] : finv[i]); }
    for(ll i = 1; i <= k; i++) { r += (mint(i).pow(n)) * finv[i] * v[k - i]; }
    return r;
  }
  mint Stirling(ll n, ll k) {
    if(n < 0 || k < 0 || n < k) { return 0; }
    mint r = 0;
    for(ll i = 0; i <= k; i++) {
      mint t = C(k, i) * (mint(i).pow(n));
      r += (k - i) & 1 ? -t : t;
    }
    return r * finv[k];
  }
} C(1000001);

const mint inv2 = mint(2).inv();

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);

  int N, Q;
  cin >> N >> Q;

  mint r = 0;
  for(int i = 1; i <= N; i++) { r += (N - i + 1) * C[i + 1]; }

  r *= (mint(N) * (N + 1) * inv2).pow(Q - 1) * Q * C(N + 1);
  cout << r.val() << "\n";
}
0