87 lines
2.3 KiB
Python
87 lines
2.3 KiB
Python
class Polynomial:
|
|
|
|
n = 0
|
|
|
|
def __init__(self, *args):
|
|
self.degree = len(args) - 1
|
|
self.coefficients = list(args)
|
|
|
|
@staticmethod
|
|
def multiply(p1, p2):
|
|
result = [0] * (p1.degree + p2.degree + 1)
|
|
f = p1.coefficients
|
|
g = p2.coefficients
|
|
for i in range(0, len(f)):
|
|
for j in range(0, len(g)):
|
|
result[i+j] += f[i] * g[j]
|
|
result = [x % int(n) for x in result]
|
|
return Polynomial(*result)
|
|
|
|
@staticmethod
|
|
def divide(p1, p2):
|
|
def inverse(x):
|
|
for i in range(1, int(n)):
|
|
r = (i * x) % int(n)
|
|
if r == 1:
|
|
break
|
|
else:
|
|
raise ZeroDivisionError
|
|
return i
|
|
if p1.degree < p2.degree:
|
|
return p1
|
|
f = p1.coefficients
|
|
g = p2.coefficients
|
|
g_lead_coef = g[-1]
|
|
g_deg = p2.degree
|
|
while len(f) >= len(g):
|
|
f_lead_coef = f[-1]
|
|
tmp_coef = f_lead_coef * inverse(g_lead_coef)
|
|
tmp_exp = len(f) - 1 - g_deg
|
|
tmp = []
|
|
for i in range(tmp_exp):
|
|
tmp.append(0)
|
|
tmp.append(tmp_coef)
|
|
tmp_poly = Polynomial(*tmp)
|
|
sub = Polynomial.multiply(p2, tmp_poly)
|
|
f = [x - y for x, y in zip(f, sub.coefficients)]
|
|
f = [x % int(n) for x in f]
|
|
while f and f[-1] == 0:
|
|
f.pop()
|
|
return Polynomial(*f)
|
|
|
|
@staticmethod
|
|
def gcd(p1, p2):
|
|
if len(p2.coefficients) == 0:
|
|
return p1
|
|
return Polynomial.gcd(p2, Polynomial.divide(p1, p2))
|
|
|
|
|
|
n = input("Enter n: ")
|
|
Polynomial.n = int(n)
|
|
|
|
c1 = [int(x) for x in input("Enter 1st polynomial (coefficients divided by space): ").split()]
|
|
c1 = [x % Polynomial.n for x in c1]
|
|
c2 = [int(x) for x in input("Enter 2nd polynomial: ").split()]
|
|
c2 = [x % Polynomial.n for x in c2]
|
|
|
|
f = Polynomial(*c1)
|
|
g = Polynomial(*c2)
|
|
|
|
ans = []
|
|
|
|
mul = Polynomial.multiply(f, g)
|
|
ans.append(mul.coefficients)
|
|
|
|
try:
|
|
div = Polynomial.divide(f, g)
|
|
ans.append(div.coefficients)
|
|
except ZeroDivisionError as e:
|
|
ans.append(e)
|
|
|
|
try:
|
|
gcd = Polynomial.gcd(f, g)
|
|
ans.append(gcd.coefficients)
|
|
except ZeroDivisionError as e:
|
|
ans.append(e)
|
|
|
|
print(ans) |