2. 两数相加
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
解题思路:
1.首先清楚题目意思:两链表元素相加,及两个链表处于相同位置的元素相加,输出最后结果
2.注意事项:链表长度(不足的补0)、是否产生进位(进位也要参与相加)
3.明确细节:
输入为两个链表l1、l2,输出为一个链表l;
需要定义的变量有:进位数carry,一个存放结果的链表result
方案一:
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
carry = 0
result = p =ListNode(0)
while l1 or l2 or carry:
carry += (l1.val if l1 else 0) + (l2.val if l2 else 0)
p.next = ListNode(carry % 10)
p = p.next
carry //= 10
l1 = l1.next if l1 else None
l2 = l2.next if l2 else None
return result.next
方案二:
迭代:
为节省空间,使用l1作为存放结果的链表,最后结果返回l1,迭代要传入参数进位carry,所以需要更改传入参数,并赋予默认值0。
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode, carry=0) -> ListNode:
if l1 == None and l2 == None and carry == 0:
return None
if l1 == None:
l1 = ListNode(0)
if l2 == None:
l2 = ListNode(0)
l1.val, carry = (l1.val + l2.val +carry) % 10, (l1.val + l2.val +carry) // 10
l1.next = self.addTwoNumbers(l1.next, l2.next, carry)
return l1