给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
这道题并不难,不过写代码之前一定先搞清楚这两种情况
- 两链表不等长
- 两链表等长
不等长时
- 我们要选择最长的链表保存结果
- 还要计算进位,和后续位的相加
等长时
- 哪个链表都可以保存
- 同样进位计算,只不过仅有一位
这样看来不等长的计算包含了等长计算。所以我们默认不等长解决
步骤:
- 比较长度,选择最长链表a保存结果过
- 两链表相加,直到最短链表b到表尾
- 将进位c和a剩下的位相加.
代码:
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
#c 保留进位
c=0
a=l1
b=l2
#找出最先到尾的链表
while a and b:
a=a.next
b=b.next
#用a保存长链表,b保存短链表;a不为空时,说明l1长,存l1;否则l2长,a存l2
if a:
a=l1
b=l2
else:
a=l2
b=l1
#保存长链表表头
v=a
#相加,直到最短链表到尾
while 1:
c,a.val=divmod(a.val+b.val+c,10)
b=b.next
#两链表相同时,同时到null,但a可能要进位加一增长链表,所以a不能过度到null(例【5】和【5】)
if not b:
break
a=a.next
#当有进位时
while c:
#a的下一位非null,下一位直接加一
if a.next:
a=a.next
c,a.val=divmod(a.val+c,10)
#当a的下一位为null,增加链表长,值为1,c=0结束
else:
a.next=ListNode(1)
c=0
return v