結果

問題 No.3391 Line up Dominoes
コンテスト
ユーザー 回転
提出日時 2025-11-28 22:21:11
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 878 bytes
コンパイル時間 383 ms
コンパイル使用メモリ 82,708 KB
実行使用メモリ 222,328 KB
最終ジャッジ日時 2025-11-28 22:22:50
合計ジャッジ時間 4,858 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other TLE * 3 MLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

def main():
    # Codonがあればなぁ
    import sys
    input = sys.stdin.readline
    from collections import defaultdict
    MOD = 998244353
    N,M,K = list(map(int,input().split()))
    A = list(map(int,input().split()))
    A.sort()

    now = defaultdict(int)
    for i in A:
        now[i] += 1
    for i in range(M-1):
        next = defaultdict(int)
        key = list(now.keys())
        l,r = 0,0
        S = 0
        for j in key:
            while(r < len(key) and j-K <= key[r] <= j+K):
                S += now[key[r]]
                S %= MOD
                r += 1
            
            while(l < len(key) and key[l] < j-K):
                S -= now[key[l]]
                S %= MOD
                l += 1
            next[j] += S
        now,next = next,now

    ans = 0
    for i in now:
        ans += now[i]
        ans %= MOD
    print(ans)
main()
0