LeetCode in Python 2. 两数相加

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。

如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。

您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例:

输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807

Solution:

注意最后一个进位是0的时候需要把它设为None,另外每次新建next节点时,初始值应设为carry而不是0,例如:5 + 7 = 2->1

# 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
        """

        head, carry = ListNode(0), 0
        l3 = prev = head
        while l1 and l2:
            sum = l1.val + l2.val + carry
            l3.val, carry = sum % 10, sum//10
            l3.next = ListNode(carry) # e.g. 5, 7 = 2->1
            prev = l3
            l1, l2, l3 = l1.next, l2.next, l3.next
            
        l4 = l1 if l1 else l2
        
        while l4:
            sum = l4.val + carry
            l3.val, carry = sum % 10, sum//10
            l3.next = ListNode(carry)
            prev = l3
            l3, l4 = l3.next, l4.next
            
        if not carry:
            prev.next = None
        
        return head

猜你喜欢

转载自www.cnblogs.com/lowkeysingsing/p/11370992.html