mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-10-06 21:59:30 +00:00
cecf43d648
* Pyupgrade to Python 3.9
* updating DIRECTORY.md
Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
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.
|
|
"""
|
|
import math
|
|
|
|
|
|
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_of_squares = sum(i * i for i in range(1, n + 1))
|
|
square_of_sum = int(math.pow(sum(range(1, n + 1)), 2))
|
|
return square_of_sum - sum_of_squares
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print(f"{solution() = }")
|