题目:给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。 请你将两个数相加,并以相同形式返回一个表示和的链表。 你可以假设除了数字 0 之外,这两个数都不会以 0 开头。
思路
我们可以用链表来模拟加法的过程。具体地,我们从两个链表的头结点开始,将对应的数字相加,并记录进位。如果某个链表到达了末尾,则其对应的数字视为 0。最后,如果两个链表都已经遍历完毕,并且最后仍有进位,那么需要在新链表的末尾添加一个值为 1 的节点。
代码实现
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution(object):
def addTwoNumbers(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
dummy = ListNode(0)
curr = dummy
carry = 0
while l1 or l2:
x = l1.val if l1 else 0
y = l2.val if l2 else 0
s = carry + x + y
carry = s // 10
curr.next = ListNode(s % 10)
curr = curr.next
if l1:
l1 = l1.next
if l2:
l2 = l2.next
if carry > 0:
curr.next = ListNode(carry)
return dummy.next
在上面的代码中,我们首先创建一个虚拟头结点 dummy 和一个指针 curr,随后依次遍历链表 l1 和 l2。具体地,我们取出对应的数字 x 和 y,并将它们与进位 carry 相加,得到当前的和 s。我们将 s 的个位数插入新链表 curr 中,并更新进位 carry。随后,我们分别将指针 l1 和 l2 向后移动一位。最后,如果存在进位 carry,那么需要在新链表的末尾添加一个值为 1 的节点。
在上面的代码中,我们使用了 Python 中的整除运算符 // 来计算进位,以及取模运算符 % 来计算当前位的数字。由于我们是从链表的头结点开始相加的,因此最终得到的链表也是按照逆序方式存储的。