beat/p-adic-numbers
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

update modp.py

Browse source
This commit is contained in:
Ata Deniz Aydin 2017-05-14 20:22:39 +03:00
parent b28a2e6438
commit 0b694c22e8
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
modp.py
View file

@ -1,8 +1,14 @@
class ModP(int): class ModP(int):
'Integers mod p, p a prime power.' 'Integers mod p, p a prime power.'
def __new__(cls, p, num): def __new__(cls, p, num):
self = int.__new__(cls, int(num) % p)
self.p = p self.p = p
return int.__new__(cls, int(num) % p) return self
def __str__(self):
return "%d (mod %d)" % (self, self.p)
def __repr__(self):
return "%d %% %d" % (self, self.p)
# arithmetic # arithmetic
def __add__(self, other): def __add__(self, other):
@ -18,6 +24,8 @@ class ModP(int):
def __rmul__(self, other): def __rmul__(self, other):
return ModP(self.p, int(other) * int(self)) return ModP(self.p, int(other) * int(self))
def __div__(self, other): def __div__(self, other):
if not isinstance(other, ModP):
other = ModP(self.p, other)
return self * other._inv() return self * other._inv()
def __rdiv__(self, other): def __rdiv__(self, other):
return other * self._inv() return other * self._inv()