[백준/BOJ] 1011번: Fly me to the Alpha Centauri(Python)

import math, sys

for _ in range(int(input())):
    x, y = map(int, sys.stdin.readline().split())
    distance = y - x
    root = math.floor(math.sqrt(distance))
    if distance <= 3:
        print(distance)
    elif root ** 2 == distance:
        print(root * 2 - 1)
    elif root ** 2 + root < distance:
        print(root * 2 + 1)
    else:
        print(root * 2)

0 ≤ x < y < 2^31이므로 sys.stdin.readline()을 이용했다. 

규칙을 머리로는 찾을 수 없어서... 먼저 손으로 구해보았다. 다음과 같은 결과를 얻을 수 있었다.

거리가 3 이하인 경우는 거리를 출력한다.
거리의 제곱근이 정수인 경우는 제곱근*2 - 1을 출력한다.
거리가 제곱근을 내림한 것의 제곱과 제곱근의 합보다 큰 경우는 제곱근*2 + 1을 출력한다.
거리가 제곱근을 내림한 것의 제곱과 제곱근의 합보다 작은 경우는 제곱근*2를 출력한다.