如何通过减少循环和条件来优化我的代码

Question

这是我的问题：

给定一个数字列表和数字“k”，返回列表中的任意两个数字是否加起来为“k”

例如，给定[1,2,5,6]其中k是 7，返回True因为 2+5 是 7。

这是我的代码； 我想要一些关于如何矢量化它的帮助。

L = [3,4,1]
k = 5
for i in L:
    for j in L:
        if i+j == k:
            print("True")
            break
    if i+j == k:
        break

Answer 1

矢量化似乎在这里没有帮助。

但是，我提出了一种新算法，将复杂度从 O(N^2) 降低到 O(N)：

L = [3,4,1]
Lset = set(L)
k = 5

for x in L:
    if k - x in Lset:
        print("True")
        break

我认为该算法是不言自明的！ :) 它基本上遍历L并且，对于每个数字，检查它的“免费”（满足x + complimentary == k所需的数字）是否存在于列表中（转换为一组以降低成员资格检查成本）。

如果您不想计算L = [1] , k = 2 ，您可以这样做：

for x in L:
    if k - x in Lset:
        if k - x == x and L.count(x) < 2:
            continue
        print("True")
        break

Answer 2

使用 itertools.combinations(L, 2) 为您提供所有 2 个长度的序列