Оптимизация раскраски плиток

plitka = []
i = 1
while True:
    if i*i <= 2000:
        plitka.append(i*i)
        i += 1
    else:
        break
    
def get_max(limit):
    selected = 1
    for a in plitka:
        if a <= limit:
            selected = a
        else:
            break
    return selected
 
Petya = int(input())
Vasya = int(input())
otdelno = get_max(Petya) + get_max(Vasya)
vmeste = get_max(Petya+Vasya)
if otdelno > vmeste:
    print('Petya leaves paint to himself')
elif otdelno == vmeste:
    print('Equal')
else:
    print('Petya gives paint to Vasya')

def paint_used(paint_avaliable):
  size = 1
  amount_of_paint = size * size
  while size * size <= paint_avaliable:
    amount_of_paint = size * size
    size += 1
  return amount_of_paint
x, y = int(input()), int(input())
if paint_used(x) + paint_used(y) == paint_used(x + y):
  print('Equal')
elif paint_used(x) + paint_used(y) > paint_used(x + y):
  print('Petya leaves paint to himself')
else:
  print('Petya gives paint to Vasya')

def used_paint(mass):
    i = 0
    a = -1
    while i*i < mass:
        a += 1
        i += 1
    return a*a
 
def compare(x, y):
    a = paint(x)
    b = paint(y)
    c = paint(x + y)
    if (a+b) < (c):
        return "Petya gives paint to Vasya"
    elif (a+b) > (c):
        return "Petya leaves paint to himself"
    elif (a+b) == c:
        return "Equal"
 
x = int(input())
y = int(input())
print(compare(x, y))

def one(a,b):
    for i in range(1,32):
        if i*i<=a and (i+1)*(i+1)>a:
            c=i*i
 
    for j in range(1,32):
        if j*j<=b and (j+1)*(j+1)>b:
            d=j*j
 
    for l in range(1,50):
        if l*l<=(a+b) and (l+1)*(l+1)>(a+b):            
            e=l*l
 
    if c+d<e:
        print("Petya gives paint to Vasya")
    elif c+d==e:
        print("Equal")
    elif c+d>e:
        print("Petya leaves paint to himself")
 
       
a,b=int(input()),int(input())
one(a,b)

a = int(input())
b = int(input())
def sq(a):
    d = 0
    while a > d * d:
        d = d + 1
    return (d - 1) * (d - 1)
def ans(a,b):
    if sq(a) + sq(b) > sq(a + b):
        print('Petya leaves paint to himself')
    elif sq(a) + sq(b) == sq(a + b):
        print('Equal')
    else:
        print('Petya gives paint to Vasya')
ans(a,b)