在Python中查找两个链表的交点的程序
假设我们有两个排序链表L1和L2,我们必须制作一个新的排序链表,其中包含这两个列表的交集。
因此,如果输入为L1=[2,4,8],L2=[3,4,8,10],则输出将为[4,8,]。
为了解决这个问题,我们将按照以下步骤进行 –
- head := 值为0的新节点
- cur := head
- while l1和l2不为空,do
- 如果l1的值<l2的值,则
- l1 := l1的下一个
- 否则当l2的值<l1的值时,则
- l2 := l2的下一个
- 否则,
- cur的下一个 := 值与l1的值相同的新节点
- l1 := l1的下一个
- l2 := l2的下一个
- cur := cur的下一个
- 如果l1的值<l2的值,则
- 返回head的下一个
让我们看以下实现以获得更好的理解 –
更多Python相关文章,请阅读: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 solve(self, l1, l2):
head = cur = ListNode(0)
while l1 and l2:
if l1.val < l2.val:
l1 = l1.next
elif l2.val < l1.val:
l2 = l2.next
else:
cur.next = ListNode(l1.val)
l1 = l1.next
l2 = l2.next
cur = cur.next
return head.next
ob = Solution()
L1 = make_list([2, 4, 8])
L2 = make_list([3, 4, 8, 10])
print_list(ob.solve(L1, L2))
输入
[2, 4, 8], [3, 4, 8, 10]
输出
[4, 8, ]