結果

問題 No.3391 Line up Dominoes
コンテスト
ユーザー 回転
提出日時 2025-11-28 22:21:11
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
WA  
実行時間 -
コード長 878 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 231 ms
コンパイル使用メモリ 95,856 KB
実行使用メモリ 306,924 KB
最終ジャッジ日時 2026-07-17 01:42:17
合計ジャッジ時間 11,043 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other WA * 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