如何在乾草堆中找到針，有更好的解決方案嗎？

Question

所以给了“针”和“这里有针，但不是针堆”

我写的

def find_needle(n,h):
    count = 0
    words = h.split(" ")
    for word in words:
        if word == n:
            count += 1
    return count

这是O（n），但想知道是否有更好的方法？也许根本不用split？在

您如何为这个案例编写测试来检查它是否处理了所有的边缘情况？在

Answer 1

您可以使用Counter

from collections import Counter

def find_needle(n,h):
    return Counter(h.split())[n]

即：

^{pr2}$

输出：

2

---

DEMO