Added power operator

1 files changed, 7 insertions, 0 deletions

diff --git a/zmodn.h b/zmodn.h
index b3b128e..fdb1ee6 100644
--- a/zmodn.h
+++ b/zmodn.h
@@ -25,10 +25,17 @@ public:
         Zmod operator+(const Zmod& z) const { return value + z.value; }
         Zmod operator-(const Zmod& z) const { return value - z.value; }
         Zmod operator*(const Zmod& z) const { return value * z.value; }
+
         Zmod operator+=(const Zmod& z) { return (*this) = value + z.value; }
         Zmod operator-=(const Zmod& z) { return (*this) = value - z.value; }
         Zmod operator*=(const Zmod& z) { return (*this) = value * z.value; }
 
+        Zmod operator^(decltype(N) z) const {
+                if (z == 0) return 1;
+                if (z % 2 == 0) return (((*this) * (*this)) ^ (z/2));
+                return (*this) * ((*this) ^ (z-1));
+        }
+
         bool operator==(const Zmod& z) const { return value == z.value; }
         bool operator!=(const Zmod& z) const { return value != z.value; }