Punts d'una corba el·líptica modular.py

from math import sqrt
N = 1997
A = 59
B = 23
coords = []
for x in range(N):
    temp = (pow(x, 3, N) + A*x + B) % N
    while (temp < 0):
        temp += N
    for i in range(N):
        if (i*i % N == temp):
            punt1 = [x, i]
            punt2 = [x, -i]
            while (punt2[1] < 0):
                punt2[1] += N
            if (punt1 not in coords):
                coords.append(punt1)
            if (punt1 not in coords):
                coords.append(punt2)
print(coords)
print("cardinalitat de y^2 = x^3 +",A,"x +",B," sobre Z/",N,"Z =", len(coords)+1)