結果

問題 No.2215 Slide Subset Sum
コンテスト
ユーザー lam6er
提出日時 2025-03-20 20:44:05
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 761 bytes
コンパイル時間 149 ms
コンパイル使用メモリ 82,684 KB
実行使用メモリ 113,820 KB
最終ジャッジ日時 2025-03-20 20:44:17
合計ジャッジ時間 8,066 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 2
other AC * 2 TLE * 1 -- * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

def main():
    import sys
    input = sys.stdin.read().split()
    idx = 0
    N = int(input[idx]); idx +=1
    M = int(input[idx]); idx +=1
    K = int(input[idx]); idx +=1
    A = list(map(lambda x: int(x)%K, input[idx:idx+N]))
    idx +=N
    
    # Precompute the sliding windows
    # For each window, compute the DP
    for k in range(N-M+1):
        current_A = A[k:k+M]
        dp = [0]*K
        dp[0] = 1  # 初始空序列
        for num in current_A:
            new_dp = dp.copy()
            for j in range(K):
                new_dp[(j + num) % K] = (new_dp[(j + num) % K] + dp[j]) % MOD
            dp = new_dp
        ans = (dp[0] -1) % MOD  # 减去空序列
        print(ans)
        
if __name__ == "__main__":
    main()
0