Получить N троек чисел Пифагора

from math import isqrt, sqrt
 
def fun(n):
    for i in range(1,1000):
        for j in range(i+1, 1000):
            a = i*i + j*j
            if isqrt(a) == sqrt(a):
                print(i, j, isqrt(a))
                n -= 1
                if not n:
                    return
 
n = int(input('n = '))
fun(n)

from heapq import heappush, heappop
 
def fun(n):
    def t1(a,b,c):
        return a-2*b+2*c, 2*a-b+2*c, 2*a-2*b+3*c
 
    def t2(a,b,c):
        return a+2*b+2*c, 2*a+b+2*c, 2*a+2*b+3*c
 
    def t3(a,b,c):
        return -a+2*b+2*c, -2*a+b+2*c, -2*a+2*b+3*c
 
    res = []
    q = []
    heappush(q, [3,4,5])
 
    while len(res) < n:
        res.append(heappop(q))
        heappush(q, sorted(t1(*res[-1])))
        heappush(q, sorted(t2(*res[-1])))
        heappush(q, sorted(t3(*res[-1])))
    return res[:n]
 
n = int(input('n = '))
print(*fun(n))

from heapq import heappush, heappop
 
def fun(n):
    def t1(a, b, c):
        return a-2*b+2*c, 2*a-b+2*c, 2*a-2*b+3*c
 
    def t2(a, b, c):
        return a+2*b+2*c, 2*a+b+2*c, 2*a+2*b+3*c
 
    def t3(a, b, c):
        return -a+2*b+2*c, -2*a+b+2*c, -2*a+2*b+3*c
 
    res = []
    q = []
    heappush(q, [3, 4, 5])
 
    while len(res) < n:
        res.append(heappop(q))
        heappush(q, sorted(t1(*res[-1])))
        heappush(q, sorted(t2(*res[-1])))
        heappush(q, sorted(t3(*res[-1])))
    return res
 
n = int(input('n = '))
print(*fun(n))

def foo(n, a=3, b=4, c=5):
    print(a, b, c)
    if n == 1:
        return
    foo(n - 1, a - 2 * b + 2 * c, 2 * a - b + 2 * c, 2 * a - 2 * b + 3 * c)
 
 
foo(10)

def foo(n, res, a=3, b=4, c=5):
    res.append(sorted([a, b, c]))
    if len(res) < n:
        A = 2*a + 2*b + 2*c
        L = A - 4*b
        foo(n, res, L-a, L+b, L+c)
        L = A
        foo(n, res, L-a, L-b, L+c)
        L = A - 4*a
        foo(n, res, L+a, L-b, L+c)
 
n = int(input('n = '))
res = []
foo(n, res)
print(*sorted(res)[:n])

def isqrt(n):
    if n<=1:
        return n,n    
    a=n//2
    while True:
        b=(a+(n//a))//2
        if abs(a-b)<1 or b>a:
            return a,b
        a=b
 
def pyth_i(n):   # n троек итерационно
    res=[]
    a=1
    b=2
    count=0
    while True:
        if a > b:
            b+=1
            a=1
        c=a*a+b*b
        u1,u2=isqrt(c)
        if u1*u2==c:
            res.append((a,b,u1))
            count+=1
            if count == n:
                return res
        a+=1
        
def pyth_r(n,a=1,b=2,res=[]):  # n троек рекурсивно
    if n==0:
        return res
    if a>b:
        return pyth_r(n,1,b+1,res)
    c=a*a+b*b
    u1,u2=isqrt(c)
    if u1*u2==c:
        return pyth_r(n-1,a+1,b,res+[(a,b,u1)])
    else:
        return pyth_r(n,a+1,b,res)
        
        
print(pyth_i(20))
print(pyth_r(20))

N = int(input('N ->'))
for i in range(1, N+1):
    print(3*i, 4*i, 5*i)

def get_pythagor(N, res=[]):
    if not N:
        return res
    return get_pythagor(N-1, res + [(3*N, 4*N, 5*N)])
 
N = int(input('N ->'))
print(*get_pythagor(N), sep='\n')