Задача про бесконечный квадрат и числа в нем

import math
n = int(input('n = '))
 
i = math.ceil(n**0.5) #находим корень из этого числа округленный в большую сторону. Это значит, что число будет однозначно находиться либо в строке с таким номером либо в столбце с таким номером, либо и в строке и в столбце с таким номером
 
center = i**2-i+1 #находим центральное число этого ряда и этого же столбца. К примеру у 4 ряда и 4 строки центральным будет число 13, у 3 - 7, у 2 - 2 и т.д
 
max_ch = i**2 #находим максимально возможное число в этом ряду и этой строке
min_ch = (i-1)**2+1 #находим минимально возможное число в этом ряду и этой строке
 
if n>center: #если наше число меньше чем центральное, то 
    
    x1=max_ch-n+1 #находим вторую неизвестную координату вычитая из максимального числа в этой строке и столбце наше число и прибавляя в нему единицу
    
    x2=i
 
else: #а если же наше число не меньше чем центральное, то
    
    x1=i
    
    x2=n-min_ch+1 #находим вторую неизвестную координату вычитая из нашего числа минимальное число в этой строке или столбце и прибавляя к нему единицу
 
if i%2==0: #проверяем i на четность, если оно четное, то номером строки будет являться второе неизвестное, а номером столбца первое
    print(x2,x1)
 
else: #иначе же все будет наоборот
    print(x1,x2)