PYTHON 3 Вася захотел сделать на Новый год особую гирлянду. В его гирлянде N цветных шариков. Вася хочет запрограммировать работу гирлянды так, чтобы шарики загорались поочередно с определённым шагом K. Так как гирлянда у Васи образует кольцо, то после N-го шарика счёт продолжается с первого шарика. Напишите, в какой последовательности будут загораться шарики на гирлянде у Васи.
Входные данные
В одной строке даны два натуральных числа N и K, не превышающие 100.
Выходные данные
Выведите N чисел в одной строке через пробел – порядок загорания шариков в гирлянде.
from collections import deque
n, k = map(int, input().split())
x, d, ssum = list(map(int, input().split())), deque(), 0
b = [(0,0) for i in range(n)]
for i in range(n):
ssum += x[i]
if i >= k :
ssum -= x[i - k]
if d[0] == i - k :
d.popleft()
while len(d) and x[d[-1]] >= x[i]:
d.pop()
d.append(i)
if i >= k - 1:
nb = (b[i-k][0] + x[d[0]] * ssum, i-k+2)
b[i] = max(b[i-1], nb, key=lambda x: x[0])
i = n - 1
d = deque()
j = b[-1][1]
d.appendleft(j)
while i !=0:
i -= 1
j1 = b[i][1]
if j-k>=j1 and j1 > 0 :
d.appendleft(j1)
j = j1
print(str(len(d)))
print(" ".join(map(str, d)))