用列表理解替换 for

Question

在这个脚本中，我同时使用了列表推导和 for。我需要用理解替换 for 循环并将这个解决方案添加到列表理解中。

怎么加

for i in k:
    count_list.append(l.count(i))

在这个块内

pairs = [int(pair/2) for pair in count_list if int(pair/2) != 0]

我的代码：

def sockMerchant(ar):
    l = ar
    k = set(l)
    count_list = []
    for i in k:
         count_list.append(l.count(i))
    pairs = [int(pair/2) for pair in count_list if int(pair/2) != 0]
    return sum(pairs)

n = int(input().strip())
ar = list(map(int, input().strip().split(' ')))
result = sockMerchant(ar)
print(result)

Answer 1

您根本不应该使用列表推导式，也不应该使用您现在拥有的 for 循环。循环效率低下；通过使用 list.count()，您将遍历整个列表 l 以获取每个唯一值，从而创建一个 O(N^2) 循环。

改用collections.Counter() object 并计算 O(N) 时间：

from collections import Counter

def sockMerchant(ar):
    counts = Counter(ar)
    return sum(count//2 for count in counts.values())

甚至

def sockMerchant(ar):
    return sum(count//2 for count in Counter(ar).values())

如果你坚持单行。

请注意，sum() 不介意这里和那里的几个 0 值，所以我删除了单个“袜子”的 if 测试。另外，我使用了// floor division operator，而不是将除以2的浮点结果转换回整数。