Kangaro.py

#Kangaro problem in hackerrank
#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

import math
import os
import random
import re
import sys

# Complete the kangaroo function below.
def kangaroo(x1, v1, x2, v2):
    for i in range(10000):
        if((x1+v1)==(x2+v2)):
            return "YES"
        x1 += v1
        x2 += v2
    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()