結果

問題 No.3391 Line up Dominoes
コンテスト
ユーザー 回転
提出日時 2025-11-28 23:04:30
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 969 bytes
コンパイル時間 3,810 ms
コンパイル使用メモリ 282,144 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-11-28 23:05:11
合計ジャッジ時間 39,478 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
#include <unordered_map>
using namespace std;
#define ll long long
#define int long long
#define INF 1e18
#define MOD 998244353

int N,M,K;
int A[505050];
signed main(){
    cin >> N >> M >> K;
    for(int i = 0;i < N;i++)cin >> A[i];
    sort(A, A+N);

    vector<int> now;
    for(int i = 0;i < N;i++)now.emplace_back(1);

    for(int i = 0;i < M-1;i++){
      vector<int> next;
      for(auto i:now)next.emplace_back(1);
      int l = 0;
      int r = 0;
      int S = 0;
      for (int j = 0;j < N;j++){
        while(r < N and A[j]-K <= A[r] and A[r] <= A[j]+K){
          S += now[r];
          S %= MOD;
          r += 1;
        }

        while (l < N and A[l] < A[j]-K){
          S -= now[l];
          S %= MOD;
          l += 1;
        }

        next[j] = S;
      }
      now.swap(next);
    }

    int ans = 0;
    for(int i = 0;i < N;i++){
      ans += now[i];
      ans %= MOD;
    }
    cout << ans << endl;

    return 0;
}
0