Pinta-ala

import math
 
t = int(input())
 
def check(a:int,b:int):
    # a = c*d
    # b = (c+1)*(d+1) = c*d + c + d + 1
    # c = a/d
    # b = a + a/d + d + 1
    # b = a*(1 + 1/d) + (d + 1)
    # b = (a/d +1)*(d + 1)
 
    # b = a + a/d + d + 1
    #  a/d + d = (b -1 -a)
    # | * d
    #   d*d - (b -1 -a)*d + a= 0
    # (-b+-sqrt(b²-4ac))/(2a)
    return has_positive_solution(1,-b+1+a,a)
    
 
def has_positive_solution(a,b,c):
    return b*b- 4*a*c>=0 and (-b +math.sqrt(b*b- 4*a*c))/(2*a)>0
 
for _ in range(t):
    A,B = map(int,input().split())
    print("YES" if check(A,B) else "NO")

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED