add_two_numbers

Author: u2

src/lib.rs

@@ -1 +1,2 @@
 pub mod first;
+pub mod second;

src/second.rs

@@ -0,0 +1,38 @@
+pub fn add_two_numbers(l1: &[usize], l2: &[usize]) -> Vec<usize> {
+    let length: usize = match l1.len() >= l2.len() {
+        true => l1.len(),
+        false => l2.len(),
+    };
+    let mut vec: Vec<usize> = Vec::with_capacity(length+1);
+    let mut a: usize;
+    let mut b: usize;
+    let mut c: usize;
+    let mut x: usize = 0;
+    for i in 0..length+1 {
+        a = if i < l1.len() { l1[i] } else { 0 };
+        b = if i < l2.len() { l2[i] } else { 0 };
+        c = a + b + x;
+        if c > 9 { x = 1; c = c - 10 } else { x = 0 };
+        if !(c == 0 && i == length) { vec.push(c); };
+    }
+    println!("vec: {:?}", vec);
+    return vec;
+}
+
+#[cfg(test)]
+mod test {
+    use super::add_two_numbers;
+
+    #[test]
+    fn test_add_two_numbers() {
+        assert_eq!(add_two_numbers(&[2,4,3], &[5,6,4]), vec![7,0,8]);
+
+        assert_eq!(add_two_numbers(&[2,2], &[5,6,4]), vec![7,8,4]);
+
+        assert_eq!(add_two_numbers(&[2,2,1], &[5,6,4]), vec![7,8,5]);
+
+        assert_eq!(add_two_numbers(&[2,2,1,3], &[5,6,4]), vec![7,8,5,3]);
+
+        assert_eq!(add_two_numbers(&[9], &[9]), vec![8,1]);
+    }
+}