update Solution 002

Author: MarkKuang1991

.gitignore

@@ -0,0 +1 @@
+.idea/**/*

solution/002.Add Two Numbers/Solution.rb

@@ -10,6 +10,7 @@
 # @param {ListNode} l1
 # @param {ListNode} l2
 # @return {ListNode}
+
 def add_two_numbers(l1, l2)
   return l2 if l1 == nil
   return l1 if l2 == nil
@@ -20,32 +21,17 @@ def add_two_numbers(l1, l2)
 
   l1 = l1.next
   l2 = l2.next
-  while !l1.nil? && !l2.nil?
-    cur_val = l1.val + l2.val + add
-    tmp.next = ListNode.new(cur_val % 10)
-    tmp = tmp.next
-    add = cur_val >= 10 ? 1 : 0
-
-    l1 = l1.next
-    l2 = l2.next
-  end
-
-  until l1.nil?
-    cur_val = l1.val + add
+  while !l1.nil? || !l2.nil? || add > 0
+    cur_val = add
+    cur_val += l1.nil? ? 0 : l1.val
+    cur_val += l2.nil? ? 0 : l2.val
     tmp.next = ListNode.new(cur_val % 10)
     tmp = tmp.next
     add = cur_val >= 10 ? 1 : 0
-    l1 = l1.next
-  end
 
-  until l2.nil?
-    cur_val = l2.val + add
-    tmp.next = ListNode.new(cur_val % 10)
-    tmp = tmp.next
-    add = l2.val + add >= 10 ? 1 : 0
-    l2 = l2.next
+    l1 = l1.nil? ? l1 : l1.next
+    l2 = l2.nil? ? l2 : l2.next
   end
 
-  tmp.next = ListNode.new(1) if add == 1
   l3
 end