leetcode 2.两数相加（python)

leetcode 2.两数相加 （python）

题目

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

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

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

示例：

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

代码

思路在代码注释里

class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None


class Solution:
    def addTwoNumbers(self, l1, l2):
        """
        :param l1: ListNode
        :param l2: ListNode
        :return: ListNode
        """
        # 把l1和l2设置为长度相同的链表
        head1 = l1
        head2 = l2
        while head1 and head2:
            prior1 = head1              # 第min(len(l1),len(l2))个节点
            prior2 = head2              # 第min(len(l1),len(l2))个节点
            head1 = head1.next
            head2 = head2.next
        while head1:                    # l1比l2长
            prior2.next = ListNode(0)
            prior2 = prior2.next
            head1 = head1.next
        while head2:
            prior1.next = ListNode(0)   # l2比l1长
            prior1 = prior1.next
            head2 = head2.next
        # 开始计算
        head1 = l1
        head2 = l2
        carry = 0       # 表示进位
        while head1 and head2:
            sum = head1.val + head2.val + carry
            head1.val = sum % 10
            carry = sum // 10
            print(carry)
            prior1 = head1              # l1最后一个节点
            head1 = head1.next
            head2 = head2.next
        if carry:                       # 最后两个数相加有进位
            prior1.next = ListNode(1)
        return l1