From a91f0e7ca07fbd176af6eb4f89d0d592a6fff620 Mon Sep 17 00:00:00 2001
From: Sanders Lin <45224617+SandersLin@users.noreply.github.com>
Date: Sat, 20 Apr 2019 00:00:40 +0800
Subject: [PATCH] Updated Euler problem 21 sol1.py

---
 project_euler/problem_21/sol1.py | 16 ++--------------
 1 file changed, 2 insertions(+), 14 deletions(-)

diff --git a/project_euler/problem_21/sol1.py b/project_euler/problem_21/sol1.py
index 6d137a7d4..da29a5c7b 100644
--- a/project_euler/problem_21/sol1.py
+++ b/project_euler/problem_21/sol1.py
@@ -24,19 +24,7 @@ def sum_of_divisors(n):
 			total += i + n//i
 		elif i == sqrt(n):
 			total += i
-
 	return total-n
 
-sums = []
-total = 0
-
-for i in xrange(1, 10000):
-	n = sum_of_divisors(i)
-	
-	if n < len(sums):
-		if sums[n-1] == i:
-			total += n + i
-
-	sums.append(n)
-
-print(total)
\ No newline at end of file
+total = [i for i in range(1,10000) if sum_of_divisors(sum_of_divisors(i)) == i and sum_of_divisors(i) != i]
+print(sum(total))