-
Notifications
You must be signed in to change notification settings - Fork 8
/
Kangaroo.py
103 lines (61 loc) · 2.82 KB
/
Kangaroo.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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# You are choreographing a circus show with various animals. For one act, you are given two kangaroos on a number line ready to jump in the positive direction (i.e, toward positive infinity).
# The first kangaroo starts at location and moves at a rate of meters per jump.
# The second kangaroo starts at location and moves at a rate of meters per jump.
# You have to figure out a way to get both kangaroos at the same location at the same time as part of the show. If it is possible, return YES, otherwise return NO.
# For example, kangaroo starts at with a jump distance and kangaroo starts at with a jump distance of . After one jump, they are both at , (, ), so our answer is YES.
# Function Description
# Complete the function kangaroo in the editor below. It should return YES if they reach the same position at the same time, or NO if they don't.
# kangaroo has the following parameter(s):
# x1, v1: integers, starting position and jump distance for kangaroo 1
# x2, v2: integers, starting position and jump distance for kangaroo 2
# Input Format
# A single line of four space-separated integers denoting the respective values of , , , and .
# Constraints
# Output Format
# Print YES if they can land on the same location at the same time; otherwise, print NO.
# Note: The two kangaroos must land at the same location after making the same number of jumps.
# Sample Input 0
# 0 3 4 2
# Sample Output 0
# YES
# Explanation 0
# The two kangaroos jump through the following sequence of locations:
# image
# From the image, it is clear that the kangaroos meet at the same location (number on the number line) after same number of jumps ( jumps), and we print YES.
# Sample Input 1
# 0 2 5 3
# Sample Output 1
# NO
# Explanation 1
# The second kangaroo has a starting location that is ahead (further to the right) of the first kangaroo's starting location (i.e., ). Because the second kangaroo moves at a faster rate (meaning ) and is already ahead of the first kangaroo, the first kangaroo will never be able to catch up. Thus, we print NO.
#!/bin/python3
import math
import os
import random
import re
import sys
# Complete the kangaroo function below.
def kangaroo(x1, v1, x2, v2):
i = 10000
counter = 0
while(i > 0):
if(x1 == x2):
return "YES"
counter = 1
break
else:
x1 += v1
x2 += v2
i -= 1
if(counter == 0):
return "NO"
if __name__ == '__main__':
fptr = open(os.environ['OUTPUT_PATH'], 'w')
x1V1X2V2 = input().split()
x1 = int(x1V1X2V2[0])
v1 = int(x1V1X2V2[1])
x2 = int(x1V1X2V2[2])
v2 = int(x1V1X2V2[3])
result = kangaroo(x1, v1, x2, v2)
fptr.write(result + '\n')
fptr.close()