結果

問題 No.324 落ちてた閉路グラフ
ユーザー ああいい
提出日時 2022-01-28 14:21:39
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 828 ms / 5,000 ms
コード長 1,330 bytes
コンパイル時間 442 ms
コンパイル使用メモリ 82,140 KB
実行使用メモリ 136,108 KB
最終ジャッジ日時 2024-12-29 09:52:48
合計ジャッジ時間 8,310 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

n,m = map(int,input().split())
w = list(map(int,input().split()))

inf = -10 ** 8

import sys
if m <= 1:
    print(0)
    exit()

"""
dp1 = [[[inf] * 2 for _ in range(m+1)] for _ in range(n)]
dp1[0][1][1] = 0
dp2 = [[[inf] * 2 for _ in range(m+1)] for _ in range(n)]
dp2[0][0][0] = 0
for i in range(n-1):
    for j in range(m+1):
        dp1[i+1][j][0] = max(dp1[i][j][0],dp1[i][j][1])
        dp2[i+1][j][0] = max(dp2[i][j][0],dp2[i][j][1])
        if j > 0:
            dp1[i+1][j][1] = max(dp1[i][j-1][1] + w[i],dp1[i][j-1][0])
            dp2[i+1][j][1] = max(dp2[i][j-1][1] + w[i],dp2[i][j-1][0])
                             
dp1[-1][m][1] += w[-1]
ans = max(dp1[-1][m][1],dp1[-1][m][0],dp2[-1][m][0],dp2[-1][m][1])
print(ans)
"""
dp1 = [[inf] * 2 for _ in range(m+1)]
dp2 = [[inf] * 2 for _ in range(m+1)]
dp1[1][1] = 0
dp2[0][0] = 0
for i in range(n-1):
    n1 = [[inf] * 2 for _ in range(m+1)]
    n2 = [[inf] * 2 for _ in range(m+1)]
    for j in range(m+1):
        n1[j][0] = max(dp1[j][0],dp1[j][1])
        n2[j][0] = max(dp2[j][0],dp2[j][1])
        if j > 0:
            n1[j][1] = max(dp1[j-1][1] + w[i],dp1[j-1][0])
            n2[j][1] = max(dp2[j-1][1] + w[i],dp2[j-1][0])
    dp1 = n1
    dp2 = n2
dp1[m][1] += w[-1]
ans = max(dp1[m][0],dp1[m][1],dp2[m][0],dp2[m][1])
print(ans)
0