2. 两数相加

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

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

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

示例 1:

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

示例 2:

输入:l1 = [0], l2 = [0]
输出:[0]

示例 3:

输入:l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出:[8,9,9,9,0,0,0,1]

提示:

  • 每个链表中的节点数在范围 [1, 100] 内
  • 0 <= Node.val <= 9
  • 题目数据保证列表表示的数字不含前导零

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

一解

分别遍历两个链表,注意细节操作即可:

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        tmp = res = ListNode()
        self.val = 0
        while l1 or l2:
            if l1 and l2:
                self.val = l1.val + l2.val + self.val//10
                l1 = l1.next
                l2 = l2.next
            elif l1:
                self.val = l1.val + self.val//10
                l1 = l1.next
            else:
                self.val = l2.val + self.val//10
                l2 = l2.next
            tmp.next = ListNode(self.val % 10)
            tmp = tmp.next
        if self.val // 10 > 0:
            tmp.next = ListNode(1)

        return res.next

上述代码可使用 val 分别计算 l1.val 和 l2.val ,简化逻辑:

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        cur = head = ListNode()
        val = carry = 0
        while carry or l1 or l2:
            val = carry
            if l1: val, l1 = l1.val + val, l1.next
            if l2: val, l2 = l2.val + val, l2.next
            carry, val = divmod(val, 10)
            cur.next = cur = ListNode(val)

        return head.next