結果

問題 No.3014 岩井満足性問題
コンテスト
ユーザー Theta
提出日時 2025-02-12 13:39:48
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,169 bytes
コンパイル時間 383 ms
コンパイル使用メモリ 82,316 KB
実行使用メモリ 274,044 KB
最終ジャッジ日時 2025-02-12 13:40:03
合計ジャッジ時間 12,494 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15 MLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import inf, isinf


def main():
    N, D, K = map(int, input().split())
    A = list(map(int, input().split()))
    C = list(map(int, input().split()))
    dp_table = [[[-inf for _ in range(K+1)]
                 for _ in range(D+1)] for _ in range(2)]
    dp_table[0][0][0] = 0
    for idx, (sa, beau) in enumerate(zip(A, C)):
        for cur_d in range(D+1):
            for cur_k in range(K+1):
                if isinf(dp_table[0][cur_d][cur_k]):
                    continue
                dp_table[1][cur_d][cur_k] = max(
                    dp_table[1][cur_d][cur_k],
                    dp_table[0][cur_d][cur_k]
                )
                if cur_d == D:
                    continue
                next_k = min(cur_k+beau, K)
                dp_table[1][cur_d+1][next_k] = max(
                    dp_table[1][cur_d+1][next_k],
                    dp_table[0][cur_d][cur_k] + sa
                )
        dp_table[0] = dp_table[1]
        dp_table[1] = [[-inf for _ in range(K+1)] for _ in range(D+1)]
    ans = dp_table[0][-1][-1]
    if isinf(ans):
        print("No")
        return
    print(ans)


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