题目
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 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]来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-two-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
1. 原始解法
建立一个新的链表存储答案。首先按位遍历l1和l2,用变量add记录是否有进位。如果两数之和大于10,则有进位。
v1 = l1.val
v2 = l2.val
v_sum = v1 + v2 + add
add = 0
if v_sum >= 10:
v_sum -= 10
add = 1
当l1或者l2被读完时,将另外一个链表剩下的元素接入存储答案的链表(仍使用加法而非直接合并)。
最后需要注意的是如果所有元素都被读完,但仍有进位,即add为1,需要在结尾再添加一个节点。
结果:通过测试用例,用时52ms,内存使用量15MB。
完整代码:
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
add = 0
answer = ListNode(0)
p = answer
while l1 and l2:
v1 = l1.val
v2 = l2.val
v_sum = v1 + v2 + add
add = 0
if v_sum >= 10:
v_sum -= 10
add = 1
p.next = ListNode(v_sum)
p = p.next
l1 = l1.next
l2 = l2.next
if l1:
while l1:
v_sum = l1.val + add
add = 0
if v_sum >= 10:
v_sum -= 10
add = 1
p.next = ListNode(v_sum)
p = p.next
l1 = l1.next
if l2:
while l2:
v_sum = l2.val + add
add = 0
if v_sum >= 10:
v_sum -= 10
add = 1
p.next = ListNode(v_sum)
p = p.next
l2 = l2.next
if add:
p.next = ListNode(1)
return answer.next
2. 第一次优化
注意到其实三段while循环有很大的重复度,可以将三者合并。
当l1或者l2有元素时,读取元素,将指针指向下一个节点,同时将另外一个元素设置为0。
结果:通过测试用例,用时52ms,内存使用量14.8MB
代码:
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
add = 0
answer = ListNode(0)
p = answer
while l1 or l2:
v1 = 0
v2 = 0
if l1:
v1 = l1.val
l1 = l1.next
if l2:
v2 = l2.val
l2 = l2.next
v_sum = v1 + v2 + add
add = 0
if v_sum >= 10:
v_sum -= 10
add = 1
p.next = ListNode(v_sum)
p = p.next
if add:
p.next = ListNode(1)
return answer.next
相关知识
1. 链表