在Python中查找两个给定链表的union的程序
假设我们有两个已排序的链表L1和L2,我们必须返回一个新的已排序的链表,这是给定的两个链表的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
- 如果L1为空,则
- 返回L2
- 如果L2为空,则
- 返回L1
- 如果L1的值小于L2的值,则
- res:= L1
- res的下一个:= solve(L1的下一个,L2)
- 否则当L2的值小于L1的值时,
- res:= L2
- res的下一个:= solve(L2的下一个,L1)
- 否则,
- res:=L1
- res的下一个:= solve(L1的下一个,L2的下一个)
- 返回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,]