Leetcode 2 两数相加

题目

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

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

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

示例：

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

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

解题思路

• 判断是否为空
• 从头开始相加，考虑进位
• 当一个链表为空，开始单独遍历另外一条
• 最后考虑，进位变量还不是空，单独存储下一位
  返回

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        if not l1:
            return l2
        if not l2:
            return l1
        
        s = 0 
        head = curr = ListNode(0)
        while l1 and l2:
            curr.next = ListNode((l1.val+l2.val+s) % 10)
            s = (l1.val+l2.val+s) // 10
            curr = curr.next
            l1 = l1.next
            l2 = l2.next
        
        if l1: 
            while l1:
                curr.next = ListNode((l1.val+s) % 10)
                s = (l1.val+s) // 10
                curr = curr.next
                l1 = l1.next
                
        if l2: 
            while l2:
                curr.next = ListNode((l2.val+s) % 10)
                s = (l2.val+s) // 10
                curr = curr.next
                l2 = l2.next
            
            
        if s == 1:
            curr.next = ListNode(1)
        return head.next