optimization for problem09 in project_euler

Author: realDuYuanChao

project_euler/problem_09/sol1.py

@@ -23,11 +23,9 @@ def solution():
     """
     for a in range(300):
         for b in range(400):
-            for c in range(500):
-                if a < b < c:
-                    if (a ** 2) + (b ** 2) == (c ** 2):
-                        if (a + b + c) == 1000:
-                            return a * b * c
+            c = 1000 - a - b
+            if a < b < c and (a ** 2) + (b ** 2) == (c ** 2):
+                return a * b * c
 
 
 if __name__ == "__main__":

project_euler/problem_09/sol3.py

@@ -25,11 +25,10 @@ def solution():
     # 31875000
     """
     return [
-        a * b * c
+        a * b * (1000 - a - b)
         for a in range(1, 999)
         for b in range(a, 999)
-        for c in range(b, 999)
-        if (a * a + b * b == c * c) and (a + b + c == 1000)
+        if (a * a + b * b == (1000 - a - b) ** 2)
     ][0]