По условию задачи, используемое деление - это деление с округлением вниз. То есть 10 * 9 / 8 = 90 / 8 = 11.

Можно легко заметить, что:

5 * 4 / 3 = 6
6 * 5 / 4 = 7
10 * 9 / 8 = 11
...

И мы можем вывести формулу, что i * (i-1) / (i-2) = i+1, когда i >= 5.

Тогда мы можем упростить вычисления следующим образом:

i * (i-1) / (i-2) + (i-3) - (i-4) * (i-5) / (i-6) + (i-7) - (i-8) * .... + остальные элементы
=   (i+1) + "(i-3)" - "(i-4) * (i-5) / (i-6)" + "(i-7)" - "(i-8) * " .... + остальные элементы
=   (i+1) + "(i-3) - (i-3)" + "(i-7) - (i-7)" +  ....  + остальные элементы
=   (i+1) + остальные элементы

Можно условно назвать каждые 4 числа цельным фрагментом, так, что из N // 4 мы можем узнать, сколько всего будет фрагментов, и тогда остальные 0, 1, 2 и 3 элемента будут влиять на наш конечный результат.

когда осталось 0 элементов: результат равен (i+1) + ... + 5 - (4*3/2) + 1, что равно i+1
когда остался 1 элемент: результат равен (i+1) + ... + 6 - (5*4/3) + 2 - 1, который равен i+2
когда осталось 2 элемента: результат равен (i+1) + ... + 7 - (6*5/4) + 3 - 2 * 1, что равно i+2
когда осталось 3 элемента: результат равен (i+1) + ... + 8 - (7*6/5) + 4 - 3 * 2 / 1, который равен i-1.

И теперь мы можем собрать итоговое решение:

def almost_factorial(n): if n <= 2: return n if n <= 4: return n + 3 if (n - 4) % 4 == 0: return n + 1 elif (n - 4) % 4 <= 2: return n + 2 else: return n - 1

[Тестовое Google] Почти факториал

Оформите Premium