PythonEx/2_AddTwoNumbers.py at master · shotai/PythonEx · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def addTwoNumbers(self, l1, l2):
        """
        :type l1: ListNode
        :type l2: ListNode
        :rtype: ListNode
        """
        if not l1 and not l2:
            return None
        head = ListNode(0)
        dummy = head
        carry = 0
        while l1 and l2:
            tmp = ListNode((l1.val+l2.val+carry)%10)
            carry = (l1.val+l2.val+carry)/10
            dummy.next = tmp
            dummy = dummy.next
            l1 = l1.next
            l2 = l2.next
        while l1:
            tmp = ListNode((l1.val+carry)%10)
            carry = (l1.val+carry)/10
            dummy.next = tmp
            dummy = dummy.next
            l1 = l1.next
        while l2:
            tmp = ListNode((l2.val+carry)%10)
            carry = (l2.val+carry)/10
            dummy.next = tmp
            dummy = dummy.next
            l2 = l2.next
        if carry>0:
            dummy.next = ListNode(carry)
            dummy = dummy.next
        return head.next