一、题目
2. 两数相加 【中等】
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照逆序 的方式存储的,并且每个节点只能存储 一位 数字。请你将两个数相加,并以相同形式返回一个表示和的链表。你可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例 1:
输入:l1= [2,4,3], l2 = [5,6,4]
输出:[7,0,8]
解释:342+ 465 = 807.
二、解题思路
对于链表相加,首先我们需要创建一个空链表,保留最终的返回链表。
针对l1、l2链表不等长的问题,需要在while循环中分别判断
最后,题目说了每一位只能存储1位数字,还需要考虑进位的情况,定义一个count用来每次加和是判断进位场景即可。
三、代码
class ListNode:
def __init__(self, val=0, next=None):
self.val = val
self.next = next
class Solution:
def addTwoNumbers(self, l1, l2):
# 创建一个结点值为 None 的头结点, dummy 和 p 指向头结点, dummy 用来保存最后返回的头结点, p 用来遍历(是随着循环前进的游标)
dummy = p = ListNode(None) # 或 ListNode()
s = 0 # 初始化进位 s 为 0
while l1 or l2 or s:
# 如果 l1 或 l2 存在, 则取l1的值 + l2的值 + s(s初始为0, 如果下面有进位1, 下次加上)
s += (l1.val if l1 else 0) + (l2.val if l2 else 0)
p.next = ListNode(s % 10) # p.next 指向新链表, 用来创建一个新的链表
# s%10是一个int类型的数字,不是链表节点,ListNode(num)表示链表节点。
p = p.next # p 向后遍历
s //= 10 # 有进位情况则取模, eg. s = 18, 18 // 10 = 1
l1 = l1.next if l1 else None # 如果l1存在, 则向后遍历, 否则为 None
l2 = l2.next if l2 else None # 如果l2存在, 则向后遍历, 否则为 None
return dummy.next # 返回 dummy 的下一个节点, 因为 dummy 指向的是空的头结点, 下一个节点才是新建链表的后序节点
if __name__ =='__main__':
solution = Solution()
l1 = ListNode(2)
l1.next = ListNode(4)
l1.next.next = ListNode(3)
l2 = ListNode(5)
l2.next = ListNode(6)
l2.next.next = ListNode(4)
print(solution.addTwoNumbers(l1, l2))