Добавлена операция вычитания

2 files changed, 64 insertions, 7 deletions

diff --git a/sem2/src/main.rs b/sem2/src/main.rs
index 34a346f..df11b2e 100644
--- a/sem2/src/main.rs
+++ b/sem2/src/main.rs
@@ -10,7 +10,13 @@ fn main() {
         Err(_) => return,
     };
     let radix = match radix.parse::<i32>() {
-        Ok(number) => number,
+        Ok(number) => {
+            if number > 36 || number < 2 {
+                println!("Система счисления должна принадлежать отрезку [2,36]");
+                return;
+            }
+            number
+        }
         Err(_) => {
             println!("Основание должно быть десятичным числом");
             return;
@@ -57,21 +63,21 @@ fn main() {
     };
 
     let mpn_sum_start = Instant::now();
-    let mpn_sum = a + b;
+    let mpn_sum = a - b;
     let mpn_sum_elapsed = mpn_sum_start.elapsed();
 
     let rug_sum_start = Instant::now();
-    let _rug_sum = a_rug + b_rug;
+    let _rug_sum = a_rug - b_rug;
     let rug_sum_elapsed = rug_sum_start.elapsed();
 
-    println!("Сумма: {}", mpn_sum);
+    println!("Разность: {}", mpn_sum);
 
     println!(
-        "Операция суммы выполнена в mpn за {} наносекунд",
+        "Операция разности выполнена в mpn за {} наносекунд",
         mpn_sum_elapsed.as_nanos()
     );
     println!(
-        "Операция суммы выполнена в rug за {} наносекунд",
+        "Операция разности выполнена в rug за {} наносекунд",
         rug_sum_elapsed.as_nanos()
     );
 }
diff --git a/sem2/src/mpn.rs b/sem2/src/mpn.rs
index ecc1d5c..24880ca 100644
--- a/sem2/src/mpn.rs
+++ b/sem2/src/mpn.rs
@@ -1,6 +1,6 @@
 use std::cmp::max;
 use std::fmt;
-use std::ops::Add;
+use std::ops::{Add, Sub};
 
 pub struct Number {
     radix: usize,
@@ -41,6 +41,21 @@ impl Number {
             None
         }
     }
+
+    fn fix_leading_zeros(mut self) -> Self {
+        let mut leading = 0;
+        for i in (1..self.digits.len()).rev() {
+            if self.digits[i] == 0 {
+                leading += 1;
+            } else {
+                break;
+            }
+        }
+        for _ in 0..leading {
+            self.digits.pop();
+        }
+        return self;
+    }
 }
 
 impl fmt::Display for Number {
@@ -92,3 +107,39 @@ impl Add for Number {
         };
     }
 }
+
+impl Sub for Number {
+    type Output = Self;
+
+    fn sub(self, other: Self) -> Self::Output {
+        let len = max(self.len(), other.len());
+        let mut digits = Vec::new();
+        let mut k = 0;
+        for i in 0..len {
+            let q1 = if i < self.len() {
+                self.digits[i] as i64
+            } else {
+                0_i64
+            };
+            let q2 = if i < other.len() {
+                other.digits[i] as i64
+            } else {
+                0_i64
+            };
+            let b = self.radix as i64;
+            let res = q1 - q2 + k;
+            let digit = (res + b) % b;
+            k = if res < 0 { -1 } else { 0 };
+            digits.push(digit as u8);
+        }
+        if k != 0 {
+            digits.push(k as u8);
+        }
+
+        return Number {
+            radix: self.radix,
+            digits,
+        }
+        .fix_leading_zeros();
+    }
+}