結果

問題 No.3391 Line up Dominoes
コンテスト
ユーザー titia
提出日時 2025-12-17 03:19:02
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 507 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 263 ms
コンパイル使用メモリ 82,412 KB
実行使用メモリ 150,676 KB
最終ジャッジ日時 2025-12-17 03:19:53
合計ジャッジ時間 45,815 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other MLE * 23
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
input = sys.stdin.readline

mod=998244353

N,M,K=map(int,input().split())
A=list(map(int,input().split()))

A.sort()

MAX=[-1]*N

ind=0
for i in range(N):
    while ind<N and A[ind]-A[i]<=K:
        ind+=1
    MAX[i]=ind

MIN=[-1]*N
ind=0
for i in range(N):
    while ind<N and A[i]-A[ind]>K:
        ind+=1
    MIN[i]=ind

DP=[1]*N

for i in range(M-1):
    S=[0]
    for x in DP:
        S.append((S[-1]+x)%mod)

    DP=[(S[MAX[j]]-S[MIN[j]])%mod for j in range(N)]

print(sum(DP)%mod)
0