Update solution.hide.py

Author: josemoracard

exercises/40-compute-robot-distance/solution.hide.py

@@ -1,20 +1,23 @@
 # Your code here
-import math
+def compute_robot_distance(movements):
+    x, y = 0, 0
 
-def compute_robot_distance(props):
-    pos = [0,0]
-    new_prop = props.split(" ")
-    for x in range(len(new_prop)):
-        if new_prop[x].upper() == 'UP':
-            pos[0]+=int(new_prop[x+1])
-        elif new_prop[x].upper() == 'DOWN':
-            pos[0]-=int(new_prop[x+1])
-        elif new_prop[x].upper() == 'LEFT':
-            pos[1]-=int(new_prop[x+1])
-        elif new_prop[x].upper() == 'RIGHT':
-            pos[1]+=int(new_prop[x+1])
-        else:
-            None
-    return (int(round(math.sqrt(pos[1]**2+pos[0]**2))))
+    for move in movements:
+        direction, steps = move.split()
+        steps = int(steps)
 
-print(compute_robot_distance("UP 5 DOWN 3 LEFT 3 RIGHT 2"))
+        if direction == "UP":
+            y += steps
+        elif direction == "DOWN":
+            y -= steps
+        elif direction == "LEFT":
+            x -= steps
+        elif direction == "RIGHT":
+            x += steps
+
+    distance = (x**2 + y**2)**0.5
+    rounded_distance = round(distance)
+
+    return rounded_distance
+
+print(compute_robot_distance(["UP 5", "DOWN 3", "LEFT 3", "RIGHT 2"]))