Помогите оптимизировать код питон(не пишите если не поможите)
Здравствуйте, надо ускорить сея программу, желательно в ответе готовый код, а не ответ нейронки,спасибо
t = int(input())
for i in range(t):
n, k = map(int, input().split())
a = list(map(int, input().split()))
srednee = sum(a) / len(a)
if n % 2 == 0:
hinata = a[:len(a) // 2]
kenma = a[len(a) // 2:]
hinata.sort(reverse=True)
kenma.sort(reverse=True)
else:
hinata = a[:len(a) // 2]
kenma = a[(len(a) // 2) + 1:]
if a[len(a) // 2] > srednee:
hinata.append(a[len(a) // 2])
else:
kenma.reverse()
kenma.append(a[len(a) // 2])
kenma.reverse()
hinata.sort(reverse=True)
kenma.sort(reverse=True)
while k > 0:
srednee = (sum(hinata) + sum(kenma)) / (len(hinata) + len(kenma))
if len(hinata) == len(kenma):
if (abs(srednee - hinata[len(hinata) - 1])) > (abs(srednee - kenma[0])):
hinata.remove(hinata[len(hinata) - 1])
k -= 1
else:
kenma.remove(kenma[0])
k -= 1
else:
if len(hinata) > len(kenma):
hinata.remove(hinata[len(hinata) - 1])
k -= 1
else:
kenma.remove(kenma[0])
k -= 1
print(sum(hinata) - sum(kenma))
t = int(input())
for i in range(t):
n, k = map(int, input().split())
a = list(map(int, input().split()))
srednee = sum(a) / len(a)
if n % 2 == 0:
hinata = a[:len(a) // 2]
kenma = a[len(a) // 2:]
hinata.sort(reverse=True)
kenma.sort(reverse=True)
else:
hinata = a[:len(a) // 2]
kenma = a[(len(a) // 2) + 1:]
if a[len(a) // 2] > srednee:
hinata.append(a[len(a) // 2])
else:
kenma.reverse()
kenma.append(a[len(a) // 2])
kenma.reverse()
hinata.sort(reverse=True)
kenma.sort(reverse=True)
while k > 0:
dlh = len(hinata)
dlk = len(kenma)
srednee = (sum(hinata) + sum(kenma)) / (dlh + dlk)
if dlh == dlk:
if (abs(srednee - hinata[dlh - 1])) > (abs(srednee - kenma[0])):
hinata.remove(hinata[dlh - 1])
k -= 1
else:
kenma.remove(kenma[0])
k -= 1
else:
if dlh > dlk:
hinata.remove(hinata[dlh - 1])
k -= 1
else:
kenma.remove(kenma[0])
k -= 1
print(sum(hinata) - sum(kenma))
Пробелы поплыли, невозможно разобраться в коде...
Задачу тоже надо знать, может тут не оптимизировать надо то что есть, а другое решение.
что должен делать данный код?