สมมติว่ามี linked list เรียงแล้ว L1 และ L2 ต้องการคืน linked list ใหม่ที่เรียงแล้วเป็น union ของทั้งสอง
ตัวอย่าง: L1 = [10,20,30,40,50,60,70], L2 = [10,30,50,80,90] ผลลัพธ์ [10,20,30,40,50,60,70,80,90]
ขั้นตอนการแก้ปัญหา
- ฟังก์ชัน
solve(L1, L2)แบบ recursive:- ถ้า
L1ว่าง: คืนL2 - ถ้า
L2ว่าง: คืนL1 - ถ้า
L1.val < L2.val:res = L1, res.next = solve(L1.next, L2) - ถ้า
L2.val < L1.val:res = L2, res.next = solve(L2.next, L1) - มิฉะนั้น (เท่ากัน):
res = L1, res.next = solve(L1.next, L2.next) - คืน
res
- ถ้า
ตัวอย่างโค้ด
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):
if not L1:
return L2
if not L2:
return L1
if L1.val < L2.val:
res = L1
res.next = self.solve(L1.next, L2)
elif L2.val < L1.val:
res = L2
res.next = self.solve(L2.next, L1)
else:
res = L1
res.next = self.solve(L1.next, L2.next)
return res
ob = Solution()
L1 = make_list([10,20,30,40,50,60,70])
L2 = make_list([10,30,50,80,90])
print_list(ob.solve(L1, L2))อินพุต
[10,20,30,40,50,60,70], [10,30,50,80,90]
เอาต์พุต
[10, 20, 30, 40, 50, 60, 70, 80, 90, ]
