Merge pull request #42 from upakarp/pythagorean_triplet

Pythagorean triplet
ademclk · Aug 23, 2022 · 5115e85 · 5115e85
2 parents 159d3df + dae4072
commit 5115e85
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/challenges/pythagorean_triplet/solution.py b/challenges/pythagorean_triplet/solution.py
@@ -0,0 +1,11 @@
+def pythagorean_triplet(num):
+    for a in range(1, num // 3 + 1):
+        for b in range(a+1, num // 2 + 1):
+            c = num - a - b
+            if a**2 + b**2 == c**2:
+                return (a, b, c)
+    return (0, 0, 0)
+
+print(pythagorean_triplet(1000))
+assert (200, 375, 425) == pythagorean_triplet(1000)
+