mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-27 15:01:08 +00:00
98e9d6bdb6
* Fix style of the first ten solutions for Project Euler - Unify the header docstring, and add reference URLs to wikipedia or similar - Fix docstrings to be properly multilined - Add newlines where appropriate - Add doctests where they were missing - Remove doctests that test for the correct solution - fix obvious spelling or grammar mistakes in comments and exception messages - Fix line endings to be UNIX. This makes two of the files seem to have changed completely - no functional changes in any of the solutions were done (except for the spelling fixes mentioned above) * Fix docstrings and main function as per Style Guide
42 lines
1021 B
Python
42 lines
1021 B
Python
"""
|
|
Project Euler Problem 6: https://projecteuler.net/problem=6
|
|
|
|
Sum square difference
|
|
|
|
The sum of the squares of the first ten natural numbers is,
|
|
1^2 + 2^2 + ... + 10^2 = 385
|
|
|
|
The square of the sum of the first ten natural numbers is,
|
|
(1 + 2 + ... + 10)^2 = 55^2 = 3025
|
|
|
|
Hence the difference between the sum of the squares of the first ten
|
|
natural numbers and the square of the sum is 3025 - 385 = 2640.
|
|
|
|
Find the difference between the sum of the squares of the first one
|
|
hundred natural numbers and the square of the sum.
|
|
"""
|
|
|
|
|
|
def solution(n: int = 100) -> int:
|
|
"""
|
|
Returns the difference between the sum of the squares of the first n
|
|
natural numbers and the square of the sum.
|
|
|
|
>>> solution(10)
|
|
2640
|
|
>>> solution(15)
|
|
13160
|
|
>>> solution(20)
|
|
41230
|
|
>>> solution(50)
|
|
1582700
|
|
"""
|
|
|
|
sum_cubes = (n * (n + 1) // 2) ** 2
|
|
sum_squares = n * (n + 1) * (2 * n + 1) // 6
|
|
return sum_cubes - sum_squares
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print(f"{solution() = }")
|