สมมติว่าเรามีลิสต์ที่เรียงแล้วสองอันคือ A และ B เราต้องรวมพวกมันและสร้างลิสต์ที่เรียงแล้วเพียงอันเดียวคือ C ขนาดของลิสต์อาจแตกต่างกัน
ตัวอย่างเช่น สมมติ A = [1,2,4,7] และ B = [1,3,4,5,6,8] ลิสต์ที่รวมแล้ว C จะเป็น [1,1,2,3,4,4,5,6,7,8]
เราจะแก้ปัญหานี้ด้วยการเรียกซ้ำ (recursion) ฟังก์ชันจะทำงานดังนี้:
- สมมติลิสต์ A และ B ของฟังก์ชัน merge()
- ถ้า 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, ]
