結果

問題 No.324 落ちてた閉路グラフ
ユーザー titiatitia
提出日時 2023-07-31 01:03:06
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 2,306 bytes
コンパイル時間 154 ms
コンパイル使用メモリ 82,132 KB
実行使用メモリ 79,980 KB
最終ジャッジ日時 2024-10-09 19:48:46
合計ジャッジ時間 7,079 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 30 RE * 4
権限があれば一括ダウンロードができます

ソースコード

diff #

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

DP0=[-1<<60]*n
# 1を使う, 直前の頂点を使う
DP1=[-1<<60]*n
# 1を使う, 直前の頂点を使わない
DP2=[-1<<60]*n
# 1を使わない, 直前の頂点を使う
DP3=[-1<<60]*n
# 1を使わない, 直前の頂点を使わない

DP0[1]=0
DP3[0]=0

for i in range(n-1):
    NDP0=[-1<<60]*n
    NDP1=[-1<<60]*n
    NDP2=[-1<<60]*n
    NDP3=[-1<<60]*n

    if i!=n-2:

        for j in range(n-1):
            if DP0[j]!=-1<<60:
                # 次の頂点を使う
                NDP0[j+1]=max(NDP0[j+1],DP0[j]+A[i])
                # 使わない
                NDP1[j]=max(NDP1[j],DP0[j])
                
            if DP1[j]!=-1<<60:
                # 次の頂点を使う
                NDP0[j+1]=max(NDP0[j+1],DP1[j])
                # 使わない
                NDP1[j]=max(NDP1[j],DP1[j])
                
            if DP2[j]!=-1<<60:
                # 次の頂点を使う
                NDP2[j+1]=max(NDP2[j+1],DP2[j]+A[i])
                # 使わない
                NDP3[j]=max(NDP3[j],DP2[j])
                
            if DP3[j]!=-1<<60:
                # 次の頂点を使う
                NDP2[j+1]=max(NDP2[j+1],DP3[j])
                # 使わない
                NDP3[j]=max(NDP3[j],DP3[j])

    else:
        for j in range(n-1):
            if DP0[j]!=-1<<60:
                # 次の頂点を使う
                NDP0[j+1]=max(NDP0[j+1],DP0[j]+A[i]+A[i+1])
                # 使わない
                NDP1[j]=max(NDP1[j],DP0[j])
                
            if DP1[j]!=-1<<60:
                # 次の頂点を使う
                NDP0[j+1]=max(NDP0[j+1],DP1[j]+A[i+1])
                # 使わない
                NDP1[j]=max(NDP1[j],DP1[j])
                
            if DP2[j]!=-1<<60:
                # 次の頂点を使う
                NDP2[j+1]=max(NDP2[j+1],DP2[j]+A[i])
                # 使わない
                NDP3[j]=max(NDP3[j],DP2[j])
                
            if DP3[j]!=-1<<60:
                # 次の頂点を使う
                NDP2[j+1]=max(NDP2[j+1],DP3[j])
                # 使わない
                NDP3[j]=max(NDP3[j],DP3[j])

    DP0=NDP0
    DP1=NDP1
    DP2=NDP2
    DP3=NDP3

print(max(DP0[m],DP1[m],DP2[m],DP3[m]))
        
    
0