Coding-Challenges/kangaroo.py at master · raj-rox/Coding-Challenges · GitHub

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
#https://www.hackerrank.com/challenges/kangaroo/problem?h_r=profile
#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the kangaroo function below.
def kangaroo(x1, v1, x2, v2):
    # if v2>=v1:
    #     return "NO"
    # else:
    #     return "YES"
    #for loop until x1>x2. if x1==x2.then print yes
    if v2==v1:
        return "NO"

    if (x2-x1)%(v2-v1)==0 and (v2<v1):
        return "YES"
    else:
        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()