Сначала хотел через цикл for, но потом вспомнил метод floor.

Part 1, not an issue.

Part 2, already mentioned.

Solution is too simple, can be done by just adding one character.

Way too easy even for an 8 kyu kata and there are some mistakes in the description.