mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-10-07 06:09:28 +00:00
23 lines
401 B
Python
23 lines
401 B
Python
from __future__ import print_function
|
|
|
|
def diagonal_sum(n):
|
|
total = 1
|
|
|
|
for i in xrange(n/2):
|
|
odd = 2*(i+1)+1
|
|
even = 2*(i+1)
|
|
total = total + 4*odd**2 - 6*even
|
|
|
|
return total
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
|
|
if len(sys.argv) == 1:
|
|
print(diagonal_sum(1001))
|
|
else:
|
|
try:
|
|
n = int(sys.argv[1])
|
|
diagonal_sum(n)
|
|
except ValueError:
|
|
print('Invalid entry - please enter a number') |