题解 | 画展布置

对了一半

def L(list1):
    l_value = 0
    for i in range(len(list1)-1):
        l_value += abs((list1[i+1])**2-int(list1[i])**2)
    return l_value

from itertools import combinations
while True:
    try:
        n, m = map(int, input().split())
        photos_values = list(map(int, input().split()))
        result = list(combinations(photos_values, m))

        results_value_l = []
        for i in result:
            sorted_ciombo = sorted(i)
            results_value_l.append(L(sorted_ciombo))

        print(min(results_value_l))

    except:
        break