Solution to Problem 28

Project Euler/Problem 28/sol1.py

@@ -0,0 +1,23 @@
+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')