【leetcode第2题—逆序链表相加的Python算法】

目录

介绍

解题思路

代码实现

---

介绍

给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。

请你将两个数相加，并以相同形式返回一个表示和的链表。

你可以假设除了数字 0 之外，这两个数都不会以 0 开头。

示例 1：

输入：l1 = [2,4,3], l2 = [5,6,4]
输出：[7,0,8]
解释：342 + 465 = 807.

示例 2：

输入：l1 = [0], l2 = [0]
输出：[0]

示例 3：

输入：l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出：[8,9,9,9,0,0,0,1]

提示：

• 每个链表中的节点数在范围 [1, 100] 内
• 0 <= Node.val <= 9
• 题目数据保证列表表示的数字不含前导零

解题思路

1. 初始化进位 carry 为 0。
2. 初始化结果链表的虚拟头节点 dummy_head 和当前节点 current。
3. 迭代遍历两个链表，直到两个链表都遍历完毕：
   • 计算当前位的和：sum = l1.val + l2.val + carry。
   • 计算进位：carry = sum // 10。
   • 创建新节点存储当前位的值：current.next = ListNode(sum % 10)。
   • 将当前节点移动到下一个节点：current = current.next。
   • 将两个链表指针向后移动：l1 = l1.next，l2 = l2.next。
4. 处理边界条件：
   • 如果两个链表长度不同，继续迭代处理剩余部分，直到所有节点都遍历完。
5. 如果最后一位有进位，需要在结果链表最后添加一个节点。

代码实现

class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

class Solution:
    def addTwoNumbers(self, l1, l2):
        carry = 0
        dummy_head = ListNode()
        current = dummy_head
        
        while l1 or l2 or carry:
            x = l1.val if l1 else 0
            y = l2.val if l2 else 0
            _sum = x + y + carry
            
            carry = _sum // 10
            current.next = ListNode(_sum % 10)
            
            current = current.next
            if l1: l1 = l1.next
            if l2: l2 = l2.next
        
        return dummy_head.next

# 测试
l1 = ListNode(2, ListNode(4, ListNode(3)))
l2 = ListNode(5, ListNode(6, ListNode(4)))

solution = Solution()
result = solution.addTwoNumbers(l1, l2)

# 输出结果链表的值
while result:
    print(result.val, end=" -> ")
    result = result.next

7 -> 0 -> 8 ->