使用 Python 列表切片和递归查找列表交集

Question

def findIntersection(list1, list2):
    if list1 == [] or list2 == []:
        return []
    elif list1[0] < list2[0]:
        return [list1[0]] + findIntersection(list[1:], list2)
    elif list1[0] > list2[0]:
        return [list2[0]] + findIntersection(list1, list2[1:])
    else:
        return ([list1[0]] + findIntersection(list1[1:], list2[1:])) 

这是我到目前为止写的代码，最终目标是找到两个列表的交集。例如， findIntersection([1,2,4], [0,2,3]) == [2] 或 findIntersection([0,2,3,5], [1,2,4,5,6 ]) == [2,5]。

我应该从哪里开始，我做错了什么。也将不胜感激对 python 切片的更多解释。谢谢。

Answer 1

你的问题是你小心返回了不在交集的元素：

elif list1[0] < list2[0]:
    return [list1[0]] + findIntersection(list[1:], list2)
elif list1[0] > list2[0]:
    return [list2[0]] + findIntersection(list1, list2[1:])
else:
    return ([list1[0]] + findIntersection(list1[1:], list2[1:]))

在这三种情况中，只有一种标识有效的交集元素，是最后一种：该元素在两个列表中。

对于其他两个，您的递归是正确的，但添加较低的元素是错误的。你必须丢弃这个元素，因为你知道它不在两个列表中。

elif list1[0] < list2[0]:
    # Drop the lowest element of list1 and recur
    return findIntersection(list[1:], list2)
elif list1[0] > list2[0]:
    # Drop the lowest element of list2 and recur
    return findIntersection(list1, list2[1:])
else:
    The element is in both lists; add it to the solution and recur.
    return ([list1[0]] + findIntersection(list1[1:], list2[1:]))

Answer 2

[i for i in list1 if i in list2]

使用列表推导。当且仅当在 list2 中找到它们时，循环遍历 list1 中的每个元素并将它们添加到新列表中。

如果你想使用切片和递归，那么你就非常接近了......你只想在交集列表中包含一个元素，如果它在两个列表中 - 这仅在最终的else 中是正确的。此外，您的代码中有错字 - list 应该是 list1。另请注意，此方法需要对输入列表进行排序（升序或降序）。

def findIntersection(list1, list2):
    if list1 == [] or list2 == []:
        return []
    elif list1[0] < list2[0]:
        return findIntersection(list1[1:], list2)
    elif list1[0] > list2[0]:
        return findIntersection(list1, list2[1:])
    else:
        return [list1[0]] + findIntersection(list1[1:], list2[1:])

list1 = [0,2,3,5]
list2 = [1,2,3,4,5,6]
print(findIntersection(list1, list2))

返回[2,3,5]