FilesExpand file tree

 master

/

sol2.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

38 lines (29 loc) · 929 Bytes

 master

/

sol2.py

Copy path

Top

File metadata and controls

• Code
• Blame

38 lines (29 loc) · 929 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

"""

Problem:

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 = 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.

"""

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 = n * (n + 1) / 2

suma **= 2

sumb = n * (n + 1) * (2 * n + 1) / 6

return int(suma - sumb)

if __name__ == "__main__":

print(solution(int(input().strip())))