Добавлена проверка на 0

1 files changed, 27 insertions, 10 deletions

diff --git a/sem2/src/mpn.rs b/sem2/src/mpn.rs
index caccc33..1cd0383 100644
--- a/sem2/src/mpn.rs
+++ b/sem2/src/mpn.rs
@@ -41,6 +41,15 @@ impl Number {
         self.digits.len()
     }
 
+    pub fn is_zero(&self) -> bool {
+        for i in &self.digits {
+            if *i != 0 {
+                return false;
+            }
+        }
+        return true;
+    }
+
     fn parse_digit(c: u8) -> Option<u8> {
         if c >= 'a' as u8 {
             Some(c - 'a' as u8 + 10_u8)
@@ -185,15 +194,19 @@ impl Mul for Number {
 }
 
 impl Number {
-    pub fn div_rem(self, other: Self) -> (Self, Self) {
+    pub fn div_rem(self, other: Self) -> Result<(Self, Self), String> {
+        if other.is_zero() {
+            return Err("Деление на 0 запрещено".to_string());
+        }
+
         if self < other {
-            return (
+            return Ok((
                 Number {
                     radix: self.radix,
                     digits: vec![0],
                 },
                 self,
-            );
+            ));
         }
 
         fn div_digit(a: &Vec<u8>, b: u8, radix: u8) -> (Vec<u8>, u8) {
@@ -210,7 +223,7 @@ impl Number {
 
         if other.len() == 1 {
             let (div, rem) = div_digit(&self.digits, other.digits[0], self.radix as u8);
-            return (
+            return Ok((
                 Number {
                     radix: self.radix,
                     digits: div,
@@ -221,7 +234,7 @@ impl Number {
                     digits: vec![rem],
                 }
                 .fix_leading_zeros(),
-            );
+            ));
         }
 
         fn mul_digit(a: &Vec<u8>, b: u8, radix: u8) -> Vec<u8> {
@@ -300,7 +313,7 @@ impl Number {
         }
         .fix_leading_zeros();
 
-        return (div, rem);
+        return Ok((div, rem));
     }
 }
 
@@ -308,8 +321,10 @@ impl Div for Number {
     type Output = Self;
 
     fn div(self, other: Self) -> Self::Output {
-        let (div, _) = self.div_rem(other);
-        return div;
+        match self.div_rem(other) {
+            Ok((div, _)) => div,
+            Err(msg) => panic!("{msg}"),
+        }
     }
 }
 
@@ -317,8 +332,10 @@ impl Rem for Number {
     type Output = Self;
 
     fn rem(self, other: Self) -> Self::Output {
-        let (_, rem) = self.div_rem(other);
-        return rem;
+        match self.div_rem(other) {
+            Ok((_, rem)) => rem,
+            Err(msg) => panic!("{msg}"),
+        }
     }
 }