Pythonで2つの並べ替えられたリストをマージする

2つの並べ替えられたリストAとBがあるとします。それらをマージして、1つの並べ替えられたリストCのみを形成する必要があります。リストのサイズは異なる場合があります。

たとえば、A=[1,2,4,7]およびB=[1,3,4,5,6,8]とすると、マージされたリストCは[1,1,2,3,4、 4,5,6,7,8]

これは再帰を使用して解決します。したがって、関数は次のように機能します-

• 関数merge（）のリストAとBを想定します
• Aが空の場合はBを返し、Bが空の場合はAを返します
• Aの値<=Bの値の場合、A.next =merge（A.next、B）そしてAを返します
• それ以外の場合は、B.next =merge（A、B.next）を実行し、Bを返します

理解を深めるために実装を見てみましょう

例（Python）

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 mergeTwoLists(self, l1, l2):
      """
      :type l1: ListNode
      :type l2: ListNode
      :rtype: ListNode
      """
      if not l1:
         return l2
      if not l2:
         return l1
      if(l1.val<=l2.val):
         l1.next = self.mergeTwoLists(l1.next,l2)
         return l1
      else:
         l2.next = self.mergeTwoLists(l1,l2.next)
         return l2
head1 = make_list([1,2,4,7])
head2 = make_list([1,3,4,5,6,8])
ob1 = Solution()
head3 = ob1.mergeTwoLists(head1,head2)
print_list(head3)

入力

head1 = make_list([1,2,4,7])
head2 = make_list([1,3,4,5,6,8])

出力

[1, 1, 2, 3, 4, 4, 5, 6, 7, 8, ]