optimization of project euler problem 10

1 files changed, 6 insertions, 6 deletions

diff --git a/src/projecteuler/010.py b/src/projecteuler/010.py
index 1a4a194..475ade8 100644
--- a/src/projecteuler/010.py
+++ b/src/projecteuler/010.py
@@ -1,18 +1,18 @@
+import math
+
 limit = 2000000
 
-number_list = [False]
-for i in range(2, limit+1):
-	number_list.append(True)
+number_list = [False] + [True]*(limit-1)
 
-for i in range(2, limit+1):
-	x = i*2
+for i in xrange(2, int(math.sqrt(limit+1))):
+	x = i*i
 	while(x <= len(number_list)):
 		number_list[x-1] = False
 		x += i
 
 sum = 0
 
-for i in range(1, limit+1):
+for i in xrange(1, limit+1):
 	if number_list[i-1]:
 		sum += i