結果

問題 No.324 落ちてた閉路グラフ
ユーザー rpy3cpprpy3cpp
提出日時 2015-12-18 22:48:32
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 168 ms / 5,000 ms
コード長 1,412 bytes
コンパイル時間 298 ms
コンパイル使用メモリ 82,192 KB
実行使用メモリ 77,056 KB
最終ジャッジ日時 2024-11-15 19:36:33
合計ジャッジ時間 4,140 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,608 KB
testcase_01 AC 39 ms
52,608 KB
testcase_02 AC 39 ms
52,224 KB
testcase_03 AC 39 ms
52,608 KB
testcase_04 AC 106 ms
71,424 KB
testcase_05 AC 76 ms
72,752 KB
testcase_06 AC 149 ms
76,240 KB
testcase_07 AC 119 ms
76,580 KB
testcase_08 AC 168 ms
76,544 KB
testcase_09 AC 127 ms
77,056 KB
testcase_10 AC 149 ms
76,688 KB
testcase_11 AC 116 ms
76,544 KB
testcase_12 AC 42 ms
58,112 KB
testcase_13 AC 41 ms
58,496 KB
testcase_14 AC 40 ms
53,760 KB
testcase_15 AC 50 ms
63,104 KB
testcase_16 AC 38 ms
52,592 KB
testcase_17 AC 39 ms
52,608 KB
testcase_18 AC 39 ms
51,968 KB
testcase_19 AC 38 ms
52,224 KB
testcase_20 AC 37 ms
52,224 KB
testcase_21 AC 38 ms
51,840 KB
testcase_22 AC 38 ms
52,096 KB
testcase_23 AC 38 ms
52,480 KB
testcase_24 AC 38 ms
52,608 KB
testcase_25 AC 38 ms
52,072 KB
testcase_26 AC 38 ms
52,096 KB
testcase_27 AC 37 ms
51,968 KB
testcase_28 AC 37 ms
52,352 KB
testcase_29 AC 37 ms
52,224 KB
testcase_30 AC 38 ms
52,480 KB
testcase_31 AC 38 ms
52,608 KB
testcase_32 AC 96 ms
76,928 KB
testcase_33 AC 155 ms
76,356 KB
testcase_34 AC 148 ms
76,540 KB
testcase_35 AC 56 ms
64,996 KB
testcase_36 AC 109 ms
76,388 KB
testcase_37 AC 47 ms
60,544 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

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

def solve(n, m, ws):
    if m <= 1:
        return 0
    f00 = [-float('inf')] * (m + 1)
    f01 = f00[:]
    f10 = f00[:]
    f11 = f00[:]
    g00 = f00[:]
    g01 = f00[:]
    g10 = f00[:]
    g11 = f00[:]
    # 初期値の設定。頂点1, 頂点2 を含むか含まないかの 4つの組み合わせに対応して初期化。
    f00[0] = 0      # 頂点1を含まない。最後尾の頂点を含まない。
    f01[1] = 0      # 頂点1を含まない。最後尾の頂点を含む。
    f10[1] = 0      # 頂点1を含む。最後尾の頂点を含まない。
    f11[2] = ws[0]  # 頂点1を含む。最後尾の頂点を含む。(最初の 2つの頂点を含むので、最初の辺を含む)
    for i, w in enumerate(ws[1:-1], 3):  # 3つ目の頂点(=2番目の辺)から順に、これを含むか否かの場合分けで計算を進めていく。
        for k in range(max(1, m + i - n), min(i, m) + 1):
            g00[k] = max(f00[k], f01[k])
            g01[k] = max(f00[k - 1], f01[k - 1] + w)
            g10[k] = max(f10[k], f11[k])
            g11[k] = max(f10[k - 1], f11[k - 1] + w)
        f00, g00 = g00, f00
        f01, g01 = g01, f01
        f10, g10 = g10, f10
        f11, g11 = g11, f11
        f00[0] = 0
    f11[m] += ws[-1]
    return max(f00[m], f01[m], f10[m], f11[m])

print(solve(n, m, ws))
0