number_line_jumps.py

문제

• https://www.hackerrank.com/challenges/kangaroo/problem?isFullScreen=true

해답 /code

#!/bin/python3

import math
import os
import random
import re
import sys

#
# Complete the 'kangaroo' function below.
#
# The function is expected to return a STRING.
# The function accepts following parameters:
#  1. INTEGER x1
#  2. INTEGER v1
#  3. INTEGER x2
#  4. INTEGER v2
#

def kangaroo(x1, v1, x2, v2):
    # compare v1, v2 if v2 > v1 -> NO
    # v1 > v2 and position_diff/(v2-v1) % %10 !=0 -> NO
    if v2 > v1 or v2 == v1:
        return 'NO'
    else:
        if (x2 - x1) % (v1-v2) == 0:
            return 'YES'
        else:
            return 'NO'


if __name__ == '__main__':
    # fptr = open(os.environ['OUTPUT_PATH'], 'w')

    first_multiple_input = input().rstrip().split()

    x1 = int(first_multiple_input[0])

    v1 = int(first_multiple_input[1])

    x2 = int(first_multiple_input[2])

    v2 = int(first_multiple_input[3])

    result = kangaroo(x1, v1, x2, v2)

    print(result)
    # fptr.write(result + '\n')
    #
    # fptr.close()