kangaroo.py

"""
Solved by: phanThASm
URL to problem: https://www.hackerrank.com/challenges/kangaroo/problem
"""

#!/bin/python3

import math
import os
import random
import re
import sys

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