Zaktualizuj 'wielomiany.py'

wielomiany.py

@@ -1,10 +1,13 @@
+import sys
+import ast
+
 class Polynomial:
     
     n = 0
 
-    def __init__(self, *args):
-        self.degree = len(args) - 1
-        self.coefficients = list(args)
+    def __init__(self, coef_list):
+        self.degree = len(coef_list) - 1
+        self.coefficients = coef_list
 
     @staticmethod
     def multiply(p1, p2):
@@ -15,7 +18,7 @@ class Polynomial:
             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)
+        return Polynomial(result)
 
     @staticmethod
     def divide(p1, p2):          
@@ -41,13 +44,13 @@ class Polynomial:
             for i in range(tmp_exp):
                 tmp.append(0)
             tmp.append(tmp_coef)
-            tmp_poly = Polynomial(*tmp)
+            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)
+        return Polynomial(f)
 
     @staticmethod
     def gcd(p1, p2):
@@ -56,16 +59,12 @@ class Polynomial:
         return Polynomial.gcd(p2, Polynomial.divide(p1, p2))
 
 
-n = input("Enter n: ")
-Polynomial.n = int(n)
+def main():
+    c1 = ast.literal_eval(sys.argv[2])
+    c2 = ast.literal_eval(sys.argv[3])
 
-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)
+    f = Polynomial(c1)
+    g = Polynomial(c2)
 
     ans = []
 
@@ -85,3 +84,7 @@ except ZeroDivisionError as e:
         ans.append(e)
 
     print(ans)
+
+if __name__ == "__main__":
+    n =  int(sys.argv[1])
+    main()