結果

問題 No.1515 Making Many Multiples
ユーザー LyricalMaestro
提出日時 2024-12-30 21:17:16
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,807 bytes
コンパイル時間 1,238 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 264,800 KB
最終ジャッジ日時 2024-12-30 21:18:16
合計ジャッジ時間 60,094 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 5 WA * 6 TLE * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/1515

MIN_VALUE = -10 ** 18

def main():
    N, K, X, Y = map(int, input().split())
    A = list(map(int, input().split()))

    dp = [[MIN_VALUE] * K for _ in range(K)]
    dp[X % K][Y % K] = 0
    dp[Y % K][X % K] = 0
    max_dp_x = [0] * K
    for x in range(K):
        m_value = MIN_VALUE
        for y in range(K):
            m_value = max(m_value, dp[x][y])
        max_dp_x[x] = m_value
    max_dp_y = [0] * K
    for y in range(K):
        m_value = MIN_VALUE
        for x in range(K):
            m_value = max(m_value, dp[x][y])
        max_dp_y[y] = m_value

    for a in A:
        updates = {}
        # 結局Aを捨てるもの
        for k_y in range(K):
            k_x_ = (- a - k_y) % K
            new_key = (k_x_, k_y) 
            updates[new_key] = dp[k_x_][k_y] + 1
        
        for k_y in range(K):
            k_x_ = (- a - k_y) % K
            new_key = (a % K, k_y)
            if new_key not in updates:
                updates[new_key] = dp[a % K][k_y]
            updates[new_key] = max(updates[new_key], dp[k_x_][k_y] + 1, max_dp_y[k_y])
        
        for k_x in range(K):
            k_y_ = (- a - k_x) % K
            new_key = (k_x, a % K)
            if new_key not in updates:
                updates[new_key] = dp[k_x][a % K]
            updates[new_key] = max(updates[new_key], dp[k_x_][k_y_] + 1, max_dp_x[k_x])

        #更新
        for key, value in updates.items():
            x, y = key
            dp[x][y] = max(dp[x][y], value)
            max_dp_x[x] = max(max_dp_x[x], dp[x][y])
            max_dp_y[y] = max(max_dp_y[y], dp[x][y])

    answer = MIN_VALUE
    for x in range(K):
        for y in range(K):
            answer = max(answer, dp[x][y])
    print(answer)



if __name__ == "__main__":
    main()
0