[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2024-10-10 05:01:19 +00:00
parent 78849a7437
commit ee5dbdda82

View File

@ -1,13 +1,14 @@
"""
This script demonstrates the implementation of the sum of squares of the first n natural numbers.
The function takes an integer n as input and returns the sum of squares
The function takes an integer n as input and returns the sum of squares
from 1 to n using the formula n(n + 1)(2n + 1) / 6. This formula computes the sum efficiently
without the need for iteration.
https://www.cuemath.com/algebra/sum-of-squares/
"""
def sum_of_squares(n: int) -> int:
"""
Implements the sum of squares formula for the first n natural numbers.
@ -25,9 +26,10 @@ def sum_of_squares(n: int) -> int:
>>> sum_of_squares(10)
385
"""
return n * (n + 1) * (2*n + 1) // 6
return n * (n + 1) * (2 * n + 1) // 6
if __name__ == "__main__":
import doctest
doctest.testmod()