-
Notifications
You must be signed in to change notification settings - Fork 0
/
catAndMouse.py
35 lines (29 loc) · 1.29 KB
/
catAndMouse.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
"""
Two cats and a mouse are at various positions on a line. You will be given their starting positions.
Your task is to determine which cat will reach the mouse first, assuming the mouse does not move and the cats travel at equal speed.
If the cats arrive at the same time, the mouse will be allowed to move and it will escape while they fight.
You are given q queries in the form of x, y, and z representing the respective positions for cats A and B, and for mouse C.
Complete the catAndMouse function to return the appropriate answer to each query, which will be printed on a new line.
If cat A catches the mouse first, print Cat A.
If cat B catches the mouse first, print Cat B.
If both cats reach the mouse at the same time, print Mouse C as the two cats fight and mouse escapes.
Example
x = 2
y = 5
z = 4
The cats are at positions 2 (Cat A) and 5 (Cat B), and the mouse is at position 4.
Cat B, at position 5 will arrive first since it is only 1 unit away while the other is 2 units away.
Return 'Cat B'.
"""
def catAndMouse(x, y, z):
distXZ = abs(x - z)
distYZ = abs(y - z)
if distXZ == distYZ:
return "Mouse C"
elif distXZ < distYZ:
return "Cat A"
else:
return "Cat B"
if __name__ == '__main__':
print(catAndMouse(1,2,3))
print(catAndMouse(1,3,2))