在列表2中查找列表1中的元素[重复]

Question

我编写了以下代码来查找列表中的元素，这些元素也在另一个列表中。但是这个算法在 Big O 中是 n 平方的，有没有更好的方法来解决这个问题？ 提前致谢

def printCommon(list1,list2):
    for i in list1:
        found = False
        for j in list2:
            if i == j:
                print i
                break
if __name__ == "__main__":
    list1 = [1,2,3,4,5]
    list2 = [9,8,7,6,5]
    printCommon(list1,list2)

Answer 1

作为一种更有效的方式，您可以使用set.intersection：

>>> set(list1).intersection(list2)
set([5])

一个基准：

from timeit import timeit

s1="""
def printCommon(list1,list2):
    for i in list1:
        found = False
        for j in list2:
            if i == j:
                #print i
                break

list1 = [1,2,3,4,5]
list2 = [9,8,7,6,5]
printCommon(list1,list2)
"""
s2="""
list1 = [1,2,3,4,5]
list2 = [9,8,7,6,5]
set(list1).intersection(set(list2))
    """

结果：

first:  0.129281997681
second :  0.0606861114502