結果

問題 No.615 集合に分けよう
ユーザー mitsuo
提出日時 2017-12-30 22:11:32
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,323 bytes
コンパイル時間 287 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 29,724 KB
最終ジャッジ日時 2024-12-21 13:32:56
合計ジャッジ時間 30,015 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 17 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):
        ming = list(groups.keys())[0]
        group1, v1, group2, v2 = groupSplit(ming)
        mindiff = (v1 + v2) - groups[ming]
        for g in list(groups.keys())[1:]:
            tgroup1, tv1, tgroup2, tv2 = groupSplit(g)
            tmindiff = (tv1 + tv2) - groups[g]
            if tmindiff < mindiff:
                ming = g
                group1 = tgroup1
                group2 = tgroup2
                v1, v2 = tv1, tv2
                mindiff = tmindiff
        del groups[tuple(ming)]
        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