Pythonのリンクリスト間でマージするプログラム

それぞれ長さmとnの2つのリンクリストL1とL2があるとすると、2つの位置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がnullでない場合は、
  • カウントがa-1と同じ場合、
    • end1：=temp
  • カウントが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, ]