主页 / 问题 / 1225746
Accepted
Danis
Asked:2021-12-30 21:49:04 +0000 UTC2021-12-30 21:49:04 +0000 UTC

如何删除重复代码？

  • 772

有一个代码可以找到从2到的所有素数n。它工作正常，但我不喜欢其中几乎有两个相同的if。如何在不增加功能时间的情况下删除它们？

def f(n):
    set_ = set()
    arr = [2, 3]
    for i in range(6, n+1, 6):
        a, b = i - 1, i + 1

        if a not in set_:
            set_.update(range(a, n+1, a))
            arr.append(a)
        if b not in set_:
            set_.update(range(b, n+1, b))
            arr.append(b)

    return arr
python

2 个回答

  1. Best Answer
    1. 或者，使用循环for：
    def f(n):
    set_ = set()
    arr = [2, 3]
    for i in range(6, n+1, 6):
        for var in (i-1, i+1):
            if var not in set_:
                set_.update(range(var, n+1, var))
                arr.append(var)
    return arr
    1. 发电机，如果你突然需要：
    def gen_f(n):
   yield 2
   yield 3
   set_ = set()
   for i in range(6, n+1, 6):
       for var in (i-1, i+1):
           if var not in set_:
               set_.update(range(var, n+1, var))
               yield var
   return None
    1. 但是，您的函数的输出与描述不完全匹配：
    n, f(n)
0 [2, 3] # 2 > 0 and 3 > 0
1 [2, 3] # 2 > 1 and 3 > 1
2 [2, 3] # 2 == 2 and 3 > 2
3 [2, 3] # 3 == 3
4 [2, 3]
5 [2, 3]
6 [2, 3, 5, 7] # 7 > 6
7 [2, 3, 5, 7] # 7 == 7
8 [2, 3, 5, 7] 
9 [2, 3, 5, 7] 
...
    1. 此代码将删除前 2 个错误输出：
    def f1(n):
    arr = []
    for m in (2, 3):
        if n < m:
            return arr
        arr.append(m)
    set_ = set()
    for i in range(6, n+1, 6):
        for var in (i-1, i+1):
            if var not in set_:
                set_.update(range(var, n+1, var))
                arr.append(var)
    return arr
    • 4

  2. 好吧，将常用操作放入函数中 - 标准重构：

    def f_update(arr, set_, x, n):
    if x not in set_:
        set_.update(range(x, n+1, x))
        arr.append(x)

def f(n):
    set_ = set()
    arr = [2, 3]
    for i in range(6, n+1, 6):
        a, b = i - 1, i + 1

        f_update(arr, set_, a, n)
        f_update(arr, set_, b, n)

    return arr
    • 3

相关问题