Start bringing-a-gun-to-a-trainer-fight

1 files changed, 14 insertions, 0 deletions

diff --git a/bringing-a-gun-to-a-trainer-fight/solution.py b/bringing-a-gun-to-a-trainer-fight/solution.py
new file mode 100644
index 0000000..36c45db
--- /dev/null
+++ b/bringing-a-gun-to-a-trainer-fight/solution.py
@@ -0,0 +1,14 @@
+import math
+
+
+def get_distance(vec):
+    return math.sqrt(vec[0]**2 + vec[1]**2)
+
+
+def solution(dimensions, your_position, trainer_position, distance):
+    # Verify that we can reach the trainer
+    dist_to_trainer = get_distance(
+        (trainer_position[0] - your_position[0], trainer_position[1] - your_position[10]))
+    if dist_to_trainer > distance:
+        # Unreachable
+        return 0