Кодинг-марафон. Задача 3.

Сталактиты свисают с потолка пещеры, а сталагмиты растут из пола.

Создайте функцию, которая определяет, представляет ли ввод «stalactites» (сталактиты) или «stalagmites» (сталагмиты). Если ввод содержит и сталактиты, и сталагмиты, верните «both» («оба»).

Ввод будет двухмерным списком, где 1 представляет кусок камня, а 0 — воздушное пространство.

Примеры

mineralFormation([   [0, 1, 0, 1],   [0, 1, 0, 1],   [0, 0, 0, 1],   [0, 0, 0, 0] ]) ➞ "stalactites"  mineralFormation([   [0, 0, 0, 0],   [0, 1, 0, 1],   [0, 1, 1, 1],   [0, 1, 1, 1] ]) ➞ "stalagmites"  mineralFormation([   [1, 0, 1, 0],   [1, 1, 0, 1],   [0, 1, 1, 1],   [0, 1, 1, 1] ]) ➞ "both" 

Решение

def mineral_formation(input):     if 1 in input[0] and 1 in input[-1]:         return 'both'     if 1 in input[0] :         return 'stalactites'     if 1 in input[-1]:         return 'stalagmites'

Марк Лутц «Изучаем Python»

Скачивайте книгу у нас в телеграм

Скачать ×