Как остановить цикл, когда в списке появляются одинаковые значения?

Question

Выше моя задача. Я должен остановить цикл, когда одни и те же элементы появляются в списке, а затем я должен суммировать их.

def squareA(a,b):
  res = []
  z = 0

  while z!= 3:

     res.insert(0,(a + z)**2)
     res.insert(0,(b - z)**2)
     z += 1

  print(res)
  res = list(dict.fromkeys(res))
  print(res)
  res = sum(res)

  return res

a = 5
b = 6
print(squareA(a,b))