用 Python 编写的查找以链表形式表示的数字的程序

pythonserver side programmingprogramming更新于 2026/2/15 5:00:17

假设我们有两个单链表 L1 和 L2，每个链表都表示一个数字，首先是最低有效数字，我们必须找到求和的链表。

因此，如果输入为 L1 = [5,6,4] L2 = [2,4,8]，则输出为 [7, 0, 3, 1, ]

要解决这个问题，我们将遵循以下步骤:

carry := 0
res := 值为 0 的新节点
curr := res
当 L1 不为空或 L2 不为空或 carry 非零时，执行
- l0_val := 如果 L1 不为空，则为 L1 的值，否则为 0
- l1_val := 如果 L2 不为空，则为 L2 的值，否则为 0
- sum_ := l0_val + l1_val
- carry := (sum_ + carry) / 10 的商
- add_val := (sum_ + carry) / 10 的余数
- curr.next := 值为 add_val 的新节点
- curr := curr 的下一个节点
- 如果 L1 不为空，则为 L1 的下一个节点，否则为 null
- 如果 L2 不为空，则为 L2 的下一个节点否则为空
curr 的下一个 := null
返回 res 的下一个

让我们看看以下实现以便更好地理解:

示例

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

def make_list(elements):
   head = ListNode(elements[0])
   for element in elements[1:]:
      ptr = head
      while ptr.next:
         ptr = ptr.next
      ptr.next = ListNode(element)
     
   return head

def print_list(head):
   ptr = head
   print('[', end = "")
   while ptr:
      print(ptr.val, end = ", ")
      ptr = ptr.next
   print(']')

class Solution:
   def solve(self, L1, L2):
      carry = 0
      res = ListNode(0)
      curr = res

      while L1 or L2 or carry:
         l0_val = L1.val if L1 else 0
         l1_val = L2.val if L2 else 0
         sum_ = l0_val + l1_val
         carry, add_val = divmod(sum_ + carry, 10)

         curr.next = ListNode(add_val)
         curr = curr.next

         L1 = L1.next if L1 else None
         L2 = L2.next if L2 else None

      curr.next = None

return res.next

ob = Solution()
L1 = make_list([5,6,4])
L2 = make_list([2,4,8])
print_list(ob.solve(L1, L2))