結果

問題 No.615 集合に分けよう
ユーザー mitsuomitsuo
提出日時 2017-12-30 21:48:49
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 967 bytes
コンパイル時間 225 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 24,672 KB
最終ジャッジ日時 2024-06-01 08:46:30
合計ジャッジ時間 4,470 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 30 ms
17,948 KB
testcase_01 AC 31 ms
11,008 KB
testcase_02 AC 31 ms
10,880 KB
testcase_03 AC 30 ms
10,880 KB
testcase_04 AC 31 ms
11,008 KB
testcase_05 AC 30 ms
10,880 KB
testcase_06 AC 30 ms
11,008 KB
testcase_07 AC 30 ms
10,880 KB
testcase_08 AC 30 ms
10,880 KB
testcase_09 WA -
testcase_10 WA -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 AC 39 ms
12,160 KB
testcase_16 TLE -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

def v(data, v = None):
    s = list(data)
    if len(s) < 2:
        return 0
    return int(s[-1]) - int(s[0])


def groupSplit(group):
    g1, g2 = group[:1], group[1:]
    v1, v2 = v(g1), v(g2)
    minv = v1 + v2
    for i in range(2, len(group)):
        tg1, tg2 = group[:i], group[i:]
        tv1, tv2 = v(tg1), v(tg2)
        if tv1 + tv2 < minv:
            v1 = tv1
            v2 = tv2
            g1 = tg1
            g2 = tg2
            minv = tv1 + tv2

    return g1, v1, g2, v2

def solve(_, k, d):
    d.sort(key=int)
    groups = {tuple([int(k) for k in d]) : v(d)}
    for _ in range(0, k - 1):
        mingroup = sorted(groups, key=lambda x: x[0])[0]
        del groups[tuple(mingroup)]
        group1, v1, group2, v2 = groupSplit(mingroup)
        groups[group1] = v1
        groups[group2] = v2
    return sum([s for s in groups.values()])

if __name__ == '__main__':
    l, k = map(int, input().split())
    print(solve(l, k, input().split()))
0