Pythonist

Представьте следующий треугольник:

    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