Представьте следующий треугольник:
1 2 3 4 5 6 7 8 9 10 ...
Создайте функцию, которая будет принимать целое число n и возвращать сумму всех чисел в n-м ряду такого треугольника.
Примечание: 1 <= n <= 1000
Примеры
row_sum(1) ➞ 1 row_sum(2) ➞ 5 row_sum(4) ➞ 34
Варианты решения
def row_sum(n):
return (n * n + 1) * n // 2
row_sum = lambda n: n * (n * n + 1) / 2
def row_sum(n):
return (n ** 3 + n) / 2