1 files changed, 30 insertions, 0 deletions
diff --git a/challenge-199/sgreen/python/ch-2.py b/challenge-199/sgreen/python/ch-2.py
new file mode 100755
index 0000000000..9536ef0e8b
--- /dev/null
+++ b/challenge-199/sgreen/python/ch-2.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+
+import sys
+from itertools import combinations
+
+
+def main(array):
+    # Get the x, y, and z values from the input
+    *array, x, y, z = array
+
+    # The solution is the number of good triplets.
+    count = 0
+
+    # Work through all combinations of positions
+    for c in combinations(range(len(array)), 3):
+        i, j, k = sorted(c)
+        # If we match the criteria, add one to the count
+        if abs(array[i] - array[j]) <= x and \
+                abs(array[j] - array[k]) <= y and \
+                abs(array[i] - array[k]) <= z:
+            count += 1
+
+    # Display the output
+    print(count)
+
+
+if __name__ == '__main__':
+    # Turn the strings into integers
+    n = [int(i) for i in sys.argv[1:]]
+    main(n)