結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 31 ms
18,204 KB
testcase_01 AC 31 ms
10,880 KB
testcase_02 AC 31 ms
11,008 KB
testcase_03 AC 31 ms
11,008 KB
testcase_04 AC 31 ms
11,008 KB
testcase_05 AC 32 ms
11,008 KB
testcase_06 AC 31 ms
11,008 KB
testcase_07 AC 31 ms
11,008 KB
testcase_08 AC 31 ms
10,880 KB
testcase_09 AC 32 ms
11,008 KB
testcase_10 AC 31 ms
11,008 KB
testcase_11 AC 32 ms
11,008 KB
testcase_12 AC 41 ms
10,880 KB
testcase_13 AC 44 ms
11,008 KB
testcase_14 AC 232 ms
11,136 KB
testcase_15 AC 40 ms
12,032 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):
        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