結果

問題 No.615 集合に分けよう
コンテスト
ユーザー mitsuo
提出日時 2017-12-30 21:48:49
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 967 bytes
コンパイル時間 317 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 29,676 KB
最終ジャッジ日時 2024-12-21 13:32:20
合計ジャッジ時間 29,349 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 11 WA * 2 RE * 4 TLE * 9
権限があれば一括ダウンロードができます

ソースコード

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