結果

問題 No.324 落ちてた閉路グラフ
ユーザー rpy3cpprpy3cpp
提出日時 2015-12-18 22:45:58
言語 Python3
(3.11.6 + numpy 1.26.0 + scipy 1.11.3)
結果
AC  
実行時間 1,947 ms / 5,000 ms
コード長 1,412 bytes
コンパイル時間 716 ms
コンパイル使用メモリ 10,920 KB
実行使用メモリ 8,992 KB
最終ジャッジ日時 2023-08-09 21:41:57
合計ジャッジ時間 17,508 ms
ジャッジサーバーID
(参考情報)
judge13 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 16 ms
8,160 KB
testcase_01 AC 15 ms
8,252 KB
testcase_02 AC 15 ms
8,272 KB
testcase_03 AC 16 ms
8,328 KB
testcase_04 AC 1,850 ms
8,328 KB
testcase_05 AC 163 ms
8,968 KB
testcase_06 AC 1,808 ms
8,504 KB
testcase_07 AC 807 ms
8,992 KB
testcase_08 AC 1,947 ms
8,680 KB
testcase_09 AC 1,004 ms
8,400 KB
testcase_10 AC 1,880 ms
8,624 KB
testcase_11 AC 1,024 ms
8,540 KB
testcase_12 AC 16 ms
8,444 KB
testcase_13 AC 15 ms
8,320 KB
testcase_14 AC 16 ms
8,280 KB
testcase_15 AC 19 ms
8,288 KB
testcase_16 AC 15 ms
8,144 KB
testcase_17 AC 15 ms
8,220 KB
testcase_18 AC 15 ms
8,168 KB
testcase_19 AC 15 ms
8,164 KB
testcase_20 AC 15 ms
8,224 KB
testcase_21 AC 14 ms
8,332 KB
testcase_22 AC 14 ms
8,144 KB
testcase_23 AC 15 ms
8,228 KB
testcase_24 AC 14 ms
7,856 KB
testcase_25 AC 16 ms
7,904 KB
testcase_26 AC 15 ms
8,172 KB
testcase_27 AC 14 ms
8,208 KB
testcase_28 AC 15 ms
8,316 KB
testcase_29 AC 15 ms
8,236 KB
testcase_30 AC 15 ms
7,952 KB
testcase_31 AC 15 ms
8,340 KB
testcase_32 AC 436 ms
8,928 KB
testcase_33 AC 1,466 ms
8,728 KB
testcase_34 AC 1,440 ms
8,736 KB
testcase_35 AC 21 ms
8,180 KB
testcase_36 AC 535 ms
8,364 KB
testcase_37 AC 17 ms
8,296 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