Python程序:查找两个给定链表的并集


假设我们有两个已排序的链表L1和L2,我们需要返回一个新的已排序链表,它是这两个给定链表的并集。

因此,如果输入类似于L1 = [10,20,30,40,50,60,70] L2 = [10,30,50,80,90],则输出将为[10, 20, 30, 40, 50, 60, 70, 80, 90]

为了解决这个问题,我们将遵循以下步骤:

  • 定义一个函数solve()。这将接收L1,L2作为参数。
  • 如果L1为空,则
    • 返回L2
  • 如果L2为空,则
    • 返回L1
  • 如果L1的值小于L2的值,则
    • res := L1
    • res的下一个节点 := solve(L1的下一个节点, L2)
  • 否则,如果L2的值小于L1的值,则
    • res := L2
    • res的下一个节点 := solve(L2的下一个节点, L1)
  • 否则,
    • res := L1
    • res的下一个节点 := solve(L1的下一个节点, L2的下一个节点)
  • 返回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):
      if not L1:
         return L2
      if not L2:
         return L1
      if L1.val < L2.val:
         res = L1
         res.next = self.solve(L1.next, L2)
      elif L2.val < L1.val:
         res = L2
         res.next = self.solve(L2.next, L1)
      else:
         res = L1
         res.next = self.solve(L1.next, L2.next)
   return res
ob = Solution()
L1 = make_list([10,20,30,40,50,60,70])
L2 = make_list([10,30,50,80,90])
print_list(ob.solve(L1, L2))

输入

[10,20,30,40,50,60,70], [10,30,50,80,90]

输出

[10, 20, 30, 40, 50, 60, 70, 80, 90, ]

更新于:2020年11月19日

377 次浏览

开启你的职业生涯

通过完成课程获得认证

开始学习
广告