给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
思路并不难,这种算法其实就跟刚开始学数学加法时候列式子计算一样,只要足够仔细,把所有可能想清楚即可。
最开始的提交代码为:
class Solution(object):
def addTwoNumbers(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
newNode=ListNode(0)
ansNode=newNode
b=0
while l1 and l2:
a=l1.val+l2.val+b
if a>9:
ansNode.next=ListNode(a%10)
b=1
else:
ansNode.next=ListNode(a)
b=0
l1=l1.next
l2=l2.next
ansNode=ansNode.next
while l1:
if l1.val+b==10:
ansNode.next=ListNode(0)
b=1
else:
ansNode.next=ListNode(b+l1.val)
b=0
l1=l1.next
ansNode=ansNode.next
while l2:
if l2.val+b==10:
ansNode.next=ListNode(0)
b=1
else:
ansNode.next=ListNode(b+l2.val)
b=0
l2=l2.next
ansNode=ansNode.next
if b>0:
ansNode.next=ListNode(b)
return newNode.next
这样写效率太低,最明显的缺点就是循环、判断太多了。那进一步简化:
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
newnode=ListNode(0)
l3=newnode
s=b=0
while l1 or l2 or b:
d=b+(l1.val if l1 else 0)+(l2.val if l2 else 0)
l3.next=ListNode(d if d<9 else d%10)
b=d//10
l1=l1.next if l1 else None
l2=l2.next if l2 else None
l3=l3.next
if b==1:
l3=ListNode(1)
return newnode.next
尽可能的在一个循环内一次性判断完。