結果

問題 No.3391 Line up Dominoes
コンテスト
ユーザー yu23578
提出日時 2025-10-19 16:18:48
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,799 bytes
コンパイル時間 1,932 ms
コンパイル使用メモリ 204,764 KB
実行使用メモリ 11,812 KB
最終ジャッジ日時 2025-11-28 20:52:02
合計ジャッジ時間 53,544 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 7 -- * 6
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

//セグ木での実装 with チャッピー
#include <bits/stdc++.h>
using namespace std;
const int MOD = 998244353;

// ========================
// Fenwick木 (BIT)
// ========================
struct Fenwick {
    int n;
    vector<int> bit;
    Fenwick(int n_): n(n_), bit(n_+1, 0) {}
    
    void update(int i, int x) {
        for (++i; i <= n; i += i & -i) {
            bit[i] += x;
            if (bit[i] >= MOD) bit[i] -= MOD;
        }
    }

    int sum(int i) { // [0, i)
        int res = 0;
        for (; i > 0; i -= i & -i) {
            res += bit[i];
            if (res >= MOD) res -= MOD;
        }
        return res;
    }

    int query(int l, int r) { // [l, r)
        int res = sum(r) - sum(l);
        if (res < 0) res += MOD;
        return res;
    }
};

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

    int N, M, K;
    cin >> N >> M >> K;
    vector<int> A(N);
    for (int i = 0; i < N; i++) cin >> A[i];

    // --- 前処理 ---
    sort(A.begin(), A.end());
    vector<pair<int,int>> D(N);
    for (int i = 0; i < N; i++) {
        int mae = lower_bound(A.begin(), A.end(), A[i] - K) - A.begin();
        int ato = lower_bound(A.begin(), A.end(), A[i] + K + 1) - A.begin();
        D[i] = {mae, ato};
    }

    // --- DP ---
    vector<int> dp(N, 1); // 初期状態（長さ1の列はすべて1通り）
    vector<int> newdp(N);

    for (int step = 1; step < M; step++) {
        Fenwick bit(N);
        for (int i = 0; i < N; i++) bit.update(i, dp[i]);
        for (int j = 0; j < N; j++)
            newdp[j] = bit.query(D[j].first, D[j].second);
        dp.swap(newdp);
    }

    // --- 答え出力 ---
    int ans = 0;
    for (int x : dp) {
        ans += x;
        if (ans >= MOD) ans -= MOD;
    }
    cout << ans << "\n";
}
0