src/bin/add-strings.rs

Author: bestgopher

src/bin/add-strings.rs

@@ -12,9 +12,7 @@ impl Solution {
         let mut vindex = v.len() - 1;
 
         loop {
-            let a = if i1 == 0 && i2 == 0 {
-                break;
-            } else if i1 == 0 && i2 > 0 {
+            let a = if i1 == 0 && i2 > 0 {
                 i2 -= 1;
                 num2[i2] - b'0' + i
             } else if i1 > 0 && i2 == 0 {
@@ -23,7 +21,7 @@ impl Solution {
             } else {
                 i2 -= 1;
                 i1 -= 1;
-                num1[i1] + num2[i2] -b'0' + i
+                num1[i1] + num2[i2] - b'0' - b'0' + i
             };
 
             if a >= 10 {
@@ -34,7 +32,16 @@ impl Solution {
                 i = 0;
             }
 
+            if i1 == 0 && i2 == 0 {
+                break;
+            }
+
+            vindex -= 1;
+        }
+
+        if i != 0 {
             vindex -= 1;
+            v[vindex] = 1;
         }
 
         v[vindex..].iter().fold(String::with_capacity((&v[vindex..]).len()), |mut x, y| {