Python程序:查找两个给定链表的并集
假设我们有两个已排序的链表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
- 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, ]
广告