백준 알고리즘 4030번 포켓볼(python)

이분 탐색 or 수학 문제이다. 일단 그냥 수학으로 풀었는데 이분 탐색으로 어떻게 풀어야 될지 모르겠다.

수학 풀이

case = 1
while 1:
    a, b = map(int, input().split())
    if a == b == 0:
        break
    cnt = i = 0
    while (i+1)*i // 2 < a:
        i += 1
    while (i+1)*i // 2 < b-1:
        t = (i+1)*i // 2 + 1
        if t**0.5 == int(t**0.5):
            cnt += 1
        i += 1
    print(f"Case {case}: {cnt}")
    case += 1