用 Python 编写程序查找两个给定链表的并集
pythonserver side programmingprogramming
假设我们有两个排序链表 L1 和 L2，我们必须返回一个新的排序链表，它是两个给定列表的并集。
因此，如果输入为 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
- re 的下一个 := solve (L2 的下一个，L1)
- 否则，
- res := L1
- re 的下一个 := 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, ]
相关文章
有用资源
python 参考教程 - 该教程包含有关 python 的更多信息:https://www.cainiaomax.com/python/
