在 Python 中查詢兩個連結串列的交集連結串列的程式


假設我們有兩個已排序的連結串列 L1 和 L2,我們需要建立一個新的已排序連結串列,其中包含這兩個連結串列的交集。

所以,如果輸入類似於 L1 = [2, 4, 8] L2 = [3, 4, 8, 10],則輸出將為 [4, 8, ]

要解決這個問題,我們將按照以下步驟進行 −

  • head := 一個值為 0 的新節點
  • cur := head
  • 當 l1 和 l2 不為空時,執行以下操作
    • 如果 l1 的值 < l2 的值,則
      • l1 := l1 的下一個值
    • 否則,當 l2 的值 < l1 的值時,則
      • l2 := l2 的下一個值
    • 否則,
      • cur 的下一個值 := 一個新節點,其值與 l1 的值相同
      • l1 := l1 的下一個值
      • l2 := l2 的下一個值
      • cur := cur 的下一個值
  • 返回 head 的下一個值

讓我們看看以下實現,以便更好地理解 −

示例

 即時演示

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):
      head = cur = ListNode(0)
      while l1 and l2:
         if l1.val < l2.val:
            l1 = l1.next
         elif l2.val < l1.val:
            l2 = l2.next
         else:
            cur.next = ListNode(l1.val)
            l1 = l1.next
            l2 = l2.next
            cur = cur.next
      return head.next
ob = Solution()
L1 = make_list([2, 4, 8])
L2 = make_list([3, 4, 8, 10])
print_list(ob.solve(L1, L2))

輸入

[2, 4, 8], [3, 4, 8, 10]

輸出

[4, 8, ]

更新時間: 19-11-2020

169 次瀏覽

開啟你的 職業 生涯

完成課程即可獲得認證

立即開始
廣告
© . All rights reserved.