結果

問題 No.3391 Line up Dominoes
コンテスト
ユーザー biotea
提出日時 2025-12-02 23:32:28
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 2,759 ms / 3,000 ms
コード長 976 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,875 ms
コンパイル使用メモリ 201,232 KB
実行使用メモリ 7,852 KB
最終ジャッジ日時 2025-12-02 23:33:55
合計ジャッジ時間 65,350 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

// #include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr ll inf = (1LL << 61);
ll dx[8] = {0, 1, 0, -1, 1, -1, -1, 1};
ll dy[8] = {-1, 0, 1, 0, 1, 1, -1, -1};
#define rep(i, n) for (ll i = 0; i < (ll)(n); ++i)
#define REP(i, init, n) for (ll i = (ll)init; i < (ll)(n); ++i)

ll M = 998244353;
ll a[10000];
ll dp[2][10000];
ll sum[10101];
int main() {
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
  ll n, m, k;
  cin >> n >> m >> k;
  rep(i, n) cin >> a[i];
  sort(a, a + n);
  ll cur = 0;
  rep(i, n) dp[cur][i] = 1;
  rep(i, n) sum[i + 1] = (sum[i] + dp[cur][i]) % M;
  vector<ll> L(n), R(n);
  rep(i, n) {
    L[i] = lower_bound(a, a + n, a[i] - k) - a;
    R[i] = upper_bound(a, a + n, a[i] + k) - a;
  }
  rep(i, m - 1) {
    rep(j, n) dp[1 - cur][j] = (sum[R[j]] + M - sum[L[j]]) % M;
    sum[0] = 0;
    rep(j, n) { sum[j + 1] = (sum[j] + dp[1 - cur][j]) % M; }
    cur = 1 - cur;
  }
  cout << sum[n] << endl;
}
0