Не могу сделать задачу (напишите код пожалуйста) довольно трудная
Вдоль аллеи растут деревья различной высоты. Их высаживали на расстояниях целого числа метров от начала аллеи, но высаживали разреженно, то есть не на каждом метре высажено по дереву. С одного из концов аллеи светит солнце так, что деревья отбрасывают тени вдоль аллеи. Ваша задача – узнать, может ли вампир с аллергией на солнечный свет безбоязненно пройти аллею. Впрочем, аллергия у вампира несильная и суммарно меньше 10 метров прогулки под солнечными лучами он переносит спокойно, а длины меньше метра в расчет вовсе не идут.
Для того, чтобы выяснить, безопасна ли аллея, вам придется сначала заполнить список, в котором отмечены места света и тени, а затем посчитать число метровых отметок, которые не прикрыты тенью.
Длина тени k на метр высоты дерева может меняться в зависимости от времени суток, поэтому мы передаем ее как аргумент функции. Знак k обозначает, в какую сторону Солнце светит. Положительный k означает, что светит от начала к концу аллеи, отрицательный – в обратную сторону. При k = 0 дерево закрывает тенью только то место, на котором оно растёт.
В качестве исходных данных вы будете работать со списком, в котором указана рассадка деревьев через каждый метр аллеи. Нулевой элемент списка отвечает началу аллеи, первый – расстоянию 1 метр от начала аллеи и т. д. Элементы списка – высоты деревьев в метрах. Участкам аллеи, на которых деревьев нет в списке сопоставлены нули. Для простоты считать, что высоты деревьев – целые, коэффициент k – тоже.
Обратите внимание: дерево высоты h может покрыть до kh + 1 метровых отметок (начало тени и конец считаются затенёнными). Часть тени, выступающую за границу аллеи, не учитывайте. См. пример.
Для решения этой задачи напишите три функции:
1) Функция make_shades(alley, k) должна вернуть список, в котором булевыми значениями отмечено, покрывает ли тень соответствующую метровую отметку.
2) Функция calculate_sunny_length(shades) должна получить список покрытия аллеи тенями и вернуть число непокрытых тенью метровых отметок
3) Функция main() запрашивает у пользователя коэффициент k и список высот деревьев вдоль аллеи в строку. В ответ печатает на экране “Обгорел”, если вампиру пришлось пройти 10 или больше метров под солнечным светом (вернее, 10 метровых отметок). Если вампиру пришлось пройти меньше 10 метров под Солнцем, напечатайте “Тени достаточно”.
Пример 1:
Ввод
print(make_shades([0, 0, 0, 4, 0, 1, 0, 0, 0, 0, 0, 2, 0, 0, 0, 3, 0], 1))
Вывод
[False, False, False, True, True, True, True, True, False, False, False, True, True, True, False, True, True]
Пример 2:
Ввод
main()
1
0 0 0 4 0 1 0 0 0 0 0 2 0 0 0 3 0
Вывод
Тени достаточно
Пример 3:
Ввод
print(calculate_sunny_length([True, True, True, True, True, True, False, False, False, True, True, True, True, True, True, True, False]))
Вывод
4
k, h = int(input()), list(map(int, input().split()))
l, s, d = len(h), set(), (1 if k > 0 else -1)
for i in range(l):
~~if h[i]: s |= set(range(i, min(l, max(-1, i + h[i] * k + d)), d))
print('Тени достаточно' if l - len(s) < 10 else 'Обгорел')
Если же делать функциями:
def make_shades(alley, k):
~~res, d = [False for _ in range(len(alley))], (1 if k > 0 else -1)
~~for i in range(len(alley)):
~~~~if alley[i] == 0: continue
~~~~for j in range(i, min(len(alley), max(-1, i + alley[i] * k + d)), d): res[j] = True
~~return res
def calculate_sunny_length(shades): return len(shades) - sum(shades)
def main():
~~k, h = int(input()), list(map(int, input().split()))
~~print('Тени достаточно' if calculate_sunny_length(make_shades(h, k)) < 10 else 'Обгорел')
main()
ну и как же без колхозного варианта от мастеров шаурмы и овощей:
def make_shades(alley,k):
res = []
i = 0
if k < 0:
alley.reverse()
while i < len(alley):
if alley[i] == 0:
res.append(False)
else:
if i+abs(k)*alley[i]+1 <= len(alley)-1:
res.extend([True] * (abs(k)*alley[i]+1))
i+=alley[i]
else:
res.extend([True] * (len(alley)-i))
break
i+=1
if k < 0:
res.reverse()
return res
def calculate_sunny_length(shades):
return len(shades) - sum(shades)
def main():
k, alley = int(input()), list(map(int, input().split()))
print('Тени достаточно' if calculate_sunny_length(make_shades(alley,k)) < 10 else 'Обгорел')
main()
