結果

問題 No.3391 Line up Dominoes
コンテスト
ユーザー hidehic0
提出日時 2025-11-28 22:08:00
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,792 bytes
コンパイル時間 3,495 ms
コンパイル使用メモリ 288,172 KB
実行使用メモリ 16,076 KB
最終ジャッジ日時 2025-11-28 22:08:10
合計ジャッジ時間 8,047 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other TLE * 1 -- * 22
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#line 1 "main.cpp"
#include <bits/stdc++.h>
using namespace std;
#include <atcoder/modint>
#include <atcoder/segtree>
using namespace atcoder;

#line 6 "/home/hidehic0/src/github.com/hidehic0/library_cpp/templates/alias.hpp"

template <typename T> using VC = std::vector<T>;
using ll = long long;
using ld = long double;
using pii = std::pair<ll, ll>;
using vi = VC<ll>;
using vvi = VC<vi>;
using vvvi = VC<vvi>;
using vb = VC<bool>;
using vvb = VC<vb>;
using vf = VC<double>;
using vvf = VC<vf>;
using vpii = VC<pii>;
using vvpii = VC<vpii>;
using si = std::set<ll>;
using spii = std::set<pii>;
using mii = std::map<ll, ll>;
const std::string upperlist = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
const std::string lowerlist = "abcdefghijklmnopqrstuvwxyz";
#define mp make_pair
#define dms << " " <<

constexpr int MOD998 = 998244353;
#line 4 "/home/hidehic0/src/github.com/hidehic0/library_cpp/templates/macro.hpp"

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rrep(i, n) for (int i = n - 1; i >= 0; i--)
#define REP(i, a, b) for (int i = (int)(a); i < (int)(b); ++i)
#define RREP(i, a, b) for (int i = (int)(b - 1); i >= (int)(a); i--)
#define all(a) (a).begin(), (a).end()

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

inline void input() {}
template <class H, class... T> void input(H &h, T &...t) {
  std::cin >> h;
  input(t...);
}
template <typename T> void input(std::vector<T> &v) {
  for (int i = 0; i < v.size(); i++) {
    std::cin >> v[i];
  }
}

inline void out() { std::cout << "\n"; }
template <class H, class... T> void out(H &h, T &...t) {
  std::cout << h << " ";
  out(t...);
}

template <typename T> void out(std::vector<T> v) {
  for (int i = 0; i < v.size(); i++) {
    std::cout << v[i] << (i + 1 == v.size() ? "\n" : " ");
  }
}

template <typename T> inline T ceil_div(T a, T b) { return (a + b - 1) / b; }
template <typename T> inline T mod_pow(T a, T n, T mod) {
  T res = 1;
  while (n) {
    if (n % 2 != 0) {
      res *= a;
      res %= mod;
    }

    a *= a;
    a %= mod;

    n >>= 1;
  }

  return res;
}
#line 9 "main.cpp"

using mint = modint998244353;

mint op(mint a, mint b) { return a + b; }
mint e() { return 0; }

int main() {
  ll N, M, K;
  cin >> N >> M >> K;
  vi A(N), L(N), R(N);
  input(A);
  sort(all(A));

  rep(i, N) {
    L[i] = lower_bound(all(A), A[i] - K) - A.begin();
    R[i] = upper_bound(all(A), A[i] + K) - A.begin();
  }

  segtree<mint, op, e> seg(VC<mint>(N, 1));

  rep(_, M - 1) {
    vector<mint> nxt(N);
    rep(i, N) { nxt[i] = seg.prod(L[i], R[i]); }
    rep(i, N) { seg.set(i, nxt[i]); }
  }

  cout << seg.all_prod().val() << "\n";
}
0