LeetCode 2. 两数相加

Description

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

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

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

示例：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/add-two-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Solution

注意：

• 进位的情况
• 一个链表长，一个链表短的情况
• 最后一个节点进位需要处理。

class Solution:
    def addTwoNumbers(self, h1: ListNode, h2: ListNode) -> ListNode:
        dummy = ListNode(-1)
        head = dummy
        tmp = 0
        while h1 or h2:
            val1 = h1.val if h1 else 0
            val2 = h2.val if h2 else 0
            val = (val1 + val2 + tmp) % 10
            tmp = (val1 + val2 + tmp) // 10
            node = ListNode(val)
            head.next = node
            head = head.next
            h1 = h1.next if h1 else None
            h2 = h2.next if h2 else None
        if tmp:
            head.next = ListNode(tmp)
        return dummy.next