Python合併連結串列程式


假設我們有兩個長度分別為m和n的連結串列L1和L2,以及兩個位置a和b。我們需要從L1中移除從第a個節點到第b個節點的節點,並在其中合併L2。

例如,如果輸入為L1 = [1,5,6,7,1,6,3,9,12],L2 = [5,7,1,6],a = 3,b = 6,則輸出為[1, 5, 6, 5, 7, 1, 6, 9, 12]

為了解決這個問題,我們將遵循以下步驟:

  • head2 := L2, temp := L2
  • 當temp有下一個節點時,執行:
    • temp := temp的下一個節點
  • tail2 := temp
  • count := 0
  • temp := L1
  • end1 := null, start3 := null
  • 當temp不為空時,執行:
    • 如果count等於a-1,則
      • end1 := temp
    • 如果count等於b+1,則
      • start3 := temp
      • 跳出迴圈
    • temp := temp的下一個節點
    • count := count + 1
  • end1的下一個節點 := head2
  • tail2的下一個節點 := start3
  • 返回L1

示例

讓我們來看下面的實現來更好地理解:

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(']')

def solve(L1, L2, a, b):
   head2 = temp = L2
   while temp.next:
      temp = temp.next
   tail2 = temp

   count = 0
   temp = L1
   end1, start3 = None, None
   while temp:
      if count == a-1:
         end1 = temp
      if count == b+1:
         start3 = temp
         break
      temp = temp.next
      count += 1

   end1.next = head2
   tail2.next = start3
   return L1

L1 = [1,5,6,7,1,6,3,9,12]
L2 = [5,7,1,6]
a = 3
b = 6
print_list(solve(make_list(L1), make_list(L2), a, b))

輸入

[1,5,6,7,1,6,3,9,12], [5,7,1,6], 3, 6

輸出

[1, 5, 6, 5, 7, 1, 6, 9, 12, ]

更新於:2021年10月5日

864 次瀏覽

開啟您的職業生涯

完成課程獲得認證

開始學習
廣告
© . All rights reserved.