Рыцари принесли принцессе подарки. Воинов было очень много, поэтому принцесса принимала рыцарей по K человек в день (в последний день их могло остаться меньше). Есть количество…

Чтобы решить задачу, давайте разобьем ее на несколько шагов. Мы имеем N рыцарей, и каждый из них принес определенное количество подарков. Принцесса принимает K рыцарей в день, и нам нужно подсчитать общее количество подарков, полученных в каждый день.

1. Понимание ввода данных:

• Первое число (N) — сколько всего рыцарей.
• Следующие N чисел — это количество подарков от каждого рыцаря.
• Последнее число (K) — сколько рыцарей может быть принято в день.

2. Алгоритм решения:

• Начнем с того, что будем обрабатывать данные по частям, принимая по K рыцарей за раз.
• Для каждого "дня" мы будем подсчитывать общее количество подарков и добавлять его в список.

3. Пример:

• Пусть у нас есть входные данные:

  N = 7
  подарки = [1, 2, 3, 4, 5, 6, 7]
  K = 3
  

4. Решение пошагово:

1. Инициализируем пустой список для хранения количества подарков каждого дня.
2. Проходим по списку подарков с шагом K.
3. Для каждой группы из K подарков (или оставшихся подарков, если их меньше K) вычисляем сумму и добавляем в список.

5. Реализация:

def count_gifts(presents, K):
    days = []
    for i in range(0, len(presents), K):
        # Сумируем подарки от K рыцарей
        day_gifts = sum(presents[i:i+K])
        days.append(day_gifts)
    return days

# Пример входных данных
N = 7
gifts = [1, 2, 3, 4, 5, 6, 7]
K = 3

# Получаем количество подарков по дням
result = count_gifts(gifts, K)
print(result)  # Вывод: [6, 15, 7]

6. Объяснение кода:

• Функция count_gifts принимает два аргумента: список с количеством подарков (presents) и число рыцарей в день (K).
• Мы используем конструкцию for i in range(0, len(presents), K), чтобы пройти по списку с шагом K.
• presents[i:i+K] выбирает подсписок из K элементов, начиная с индекса i. Если элементов остается меньше K (как в последней итерации), Python сам отберет только доступные элементы.
• Используя встроенную функцию sum, мы суммируем количество подарков и добавляем результат в список days.
• В конце функция возвращает список с результатами.

Таким образом, мы можем подсчитывать подарки, которые принцесса получает за каждый день, принимая по K рыцарей одновременно.