for i, v in enumerate(diff):
    acc += (i + 1) * (K - (i + 1)) * v
return acc

def closest(a, K):
    a.sort()
N = len(a)
diff = [(a[i + 1] - a[i]) for i in range(N - 1)]
min_ind = 0
min_val = closeness(diff[0:K - 1], K)
for ind in range(1, N - K + 1):
    cl = closeness(diff[ind:ind + K - 1], K)
if cl < min_val:
    min_ind = ind
min_val = cl
return a[min_ind:min_ind + K]
Найти подмножество с элементами K, которые находятся ближе всего друг к другу