結果

問題 No.3391 Line up Dominoes
コンテスト
ユーザー urunea
提出日時 2025-12-02 02:30:41
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 762 ms / 3,000 ms
コード長 1,503 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,210 ms
コンパイル使用メモリ 286,796 KB
実行使用メモリ 7,720 KB
最終ジャッジ日時 2025-12-02 02:31:00
合計ジャッジ時間 19,281 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;

static const long long MOD = 998244353;

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

    int N, M;
    long long K;
    cin >> N >> M >> K;

    vector<long long> A(N);
    for (int i = 0; i < N; i++) cin >> A[i];
    sort(A.begin(), A.end());

    // mA[j]: A[j] のときに選べる最小 index
    // MA[j]: A[j] のときに選べる最大 index
    vector<int> mA(N), MA(N);
    for (int i = 0; i < N; i++) {
        long long val = A[i];
        int l = lower_bound(A.begin(), A.end(), val - K) - A.begin();
        int r = upper_bound(A.begin(), A.end(), val + K) - A.begin() - 1; // right inclusive
        mA[i] = l;
        MA[i] = r;
    }

    vector<long long> dp(N, 1), new_dp(N, 0);

    for (int i = 0; i < M - 1; i++) {
        // 累積和 acc_dp[j] = dp[0] + dp[1] + ... + dp[j]
        vector<long long> acc_dp(N);
        acc_dp[0] = dp[0];
        for (int j = 1; j < N; j++) {
            acc_dp[j] = (acc_dp[j - 1] + dp[j]) % MOD;
        }

        for (int j = 0; j < N; j++) {
            int l = mA[j];
            int r = MA[j];
            if (l == 0) {
                new_dp[j] = acc_dp[r];
            } else {
                new_dp[j] = (acc_dp[r] - acc_dp[l - 1] + MOD) % MOD;
            }
        }

        dp = new_dp;
        fill(new_dp.begin(), new_dp.end(), 0LL);
    }

    long long ans = 0;
    for (long long x : dp) ans = (ans + x) % MOD;
    cout << ans << "\n";

    return 0;
}
0