Python/project_euler/problem_06/sol1.py

42 lines
957 B
Python
Raw Normal View History

"""
2018-10-19 12:48:28 +00:00
Problem:
2018-10-19 12:48:28 +00:00
The sum of the squares of the first ten natural numbers is,
1^2 + 2^2 + ... + 10^2 = 385
2018-10-19 12:48:28 +00:00
The square of the sum of the first ten natural numbers is,
(1 + 2 + ... + 10)^2 = 552 = 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 N natural
numbers and the square of the sum.
"""
2019-10-05 05:14:13 +00:00
def solution(n):
"""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
"""
suma = 0
sumb = 0
for i in range(1, n + 1):
suma += i ** 2
sumb += i
sum = sumb ** 2 - suma
return sum
if __name__ == "__main__":
print(solution(int(input().strip())))