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 次瀏覽

開啟你的職業生涯

透過完成課程獲得認證

開始學習
廣告