Найдите сумму первых N членов ряда 5, 11, 19, 29, 41, . . .

Учитывая целое число N . Задача состоит в том, чтобы найти сумму первых N членов ряда 5, 11, 19, 29, 41, . . . до N-го срока .

Примеры:

Input:  N = 5
Output: 105
Explanation: 5 + 11 + 19 + 29 + 41 = 105.

Input: N = 2
Output: 16
Explanation: The terms are 5 and 11

Подход: Из данного ряда сначала определить N-й член:

1st term = 5 = 1 + 4 = 1 + 2²
2nd term = 11 = 2 + 9 = 2 + 3²
3rd term = 19 = 3 + 16 = 3 + 4²
4th term = 29 = 4 + 25 = 4 + 5²
.
.
Nth term = N + (N+1)²

Таким образом, N -й член можно записать как: T _N = N + (N+1) ²

Поэтому сумма до N членов становится

1 + 2² + 2 + 3² + 3 + 4² + . . . + N + (N+1)²
= [1 + 2 + 3 + . . . + N] + [2² + 3² + 4² + . . . + (N+1)²]
= (N*(N+1))/2 + [(N+1)*(N+2)*(2*N + 3)]/6 – 1
= [N*(N+2)*(N+4)]/3

Поэтому сумма первых N членов может быть представлена как: S _N = [N*(N+2)*(N+4)]/3

Иллюстрация:

For example, take N = 5
The output will be 105.
Use N = 5, then N*(N+2)*(N+4)/3
= 5 * 7 * 9/3 = 5 * 7 * 3 = 105.
This is same as 5 + 11 + 19 + 29 + 41

Ниже приведена реализация описанного выше подхода.

Временная сложность: O(1)
Вспомогательное пространство: O(1), так как дополнительное пространство не занято.