feat: elementary school addition/multiplication

0xWOLAND · Nov 19, 2023 · fdcb457 · fdcb457
1 parent 6e9f39c
commit fdcb457
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/ntt.rs b/src/ntt.rs
@@ -132,8 +132,16 @@ fn fft(inp: Vec<BigInt>, c: &Constants, w: BigInt) -> Vec<BigInt> {
                 .zip(hi)
                 .enumerate()
                 .for_each(|(idx, (lo, hi))| {
-                    let neg = (*lo).sub_mod(*hi, MOD);
-                    *lo = (*lo).add_mod(*hi, MOD);
+                    let neg = if *lo < *hi {
+                        (MOD + *lo) - *hi
+                    } else {
+                        *lo - *hi
+                    };
+                    *lo = if *lo + *hi >= MOD {
+                        (*lo + *hi) - MOD
+                    } else {
+                        *lo + *hi
+                    };
                     *hi = neg.mul_mod(pre[nchunks * idx], MOD);
                 });
         });