2. 两数相加
- 给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
- 如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
- 您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
思路:
- 不断遍历两个链表,每次遍历都将链表a和链表b的值相加,再赋给链表a。如果有进位则记录一个进位标志。
- 循环的条件是链表a不为空或者链表b不为空,这样当整个循环结束时,链表就被串起来了。
- 注意:当循环结束时,如果进位标志为1还需要处理下边界条件。
代码实现
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
a, b = l1, l2
# 定义主指针,指向l1,用来处理
# 1. 循环结束时,进位标志为1
# 2. a,b链表长度不一致
# 的情况,最终返回l1
p = None
# 定义进位标志
carry = 0
while a or b:
# 将链表a,b的值相加,并加上可能的进位
val = (a.val if a else 0) + (b.val if b else 0) + carry
# 若val>10说明要进位,carry = 1否则为0
carry = 1 if val>=10 else 0
# 不论是否进位,val都取个位数的值
val = val % 10
# 主指针p指向a,值为val
p, p.val = a if a else b, val
# a,b向后移动,若到都头,循环结束
a, b = a.next if a else None, b.next if b else None
# 判断主指针的下一个指向,若a,b都为空,则指向空,若a,b有一个不为空,则指向不为空者
p.next = a if a else b
# 处理循环结束时,进位标志为1的情况
if carry:
p.next = ListNode(1)
return l1