leetcode-两数相加

题目描述

给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。

请你将两个数相加，并以相同形式返回一个表示和的链表。

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

 输入：l1 = [2,4,3], l2 = [5,6,4]
 输出：[7,0,8]
 解释：342 + 465 = 807.

算法思想

• 将两个链表看成是相同长度的进行遍历，如果一个链表较短则在前面补 0
• 每一位计算的同时需要考虑上一位的进位问题，而当前位计算结束后同样需要更新进位值
• 如果两个链表全部遍历完毕后，进位值为 1，则在新链表最前方添加节点 1
• trick：对于链表问题，返回结果为头结点时，通常需要先初始化一个预先指针 pre，该指针的下一个节点指向真正的头结点head

代码实现

    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:

        # 进位初始为0
        carry = 0
        head = ListNode(-1)
        cur = head

        while(l1 or l2):
            n1 = l1.val if l1 else 0
            n2 = l2.val if l2 else 0

            sum = n1 + n2 + carry

            tmp = ListNode(sum%10)
            cur.next = tmp

            cur = tmp

            carry = sum / 10
            if l1:
                l1 = l1.next
            if l2:
                l2 = l2.next

        if carry>0:
            cur.next = ListNode(carry)

        return head.next