给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 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