如何使我的代码更高效，使其运行时少于1秒

Question

我最近有一次考试（现在已经结束了），我有一个问题： question

我可以回答这个问题。然而，最后有一个测试用例说，如果该用例的运行时间为<；1s。然而，我的问题是，我无法使运行时低于1秒。下面是我的代码和测试用例。我还要补充一点，我们不允许导入任何软件包

测试用例：

longList = [i for i in range(100000)]
print(calculateAreas(longList,longList[::-1]))

我的代码：

def calculateAreas(w_list,h_list):
    width_length = len(w_list)
    height_length = len(h_list)
    list_of_areas = []
    red_area = 0
    white_area = 0
    yellow_area = 0
    for i in range(width_length):
        for n in range(height_length):
            if (i+n-2) % 3 == 0:
                red_area += (w_list[i] * h_list[n])
            if (i+n-2) % 3 == 1:
                white_area += (w_list[i] * h_list[n])
            if (i+n-2) % 3 == 2:
                yellow_area += (w_list[i] * h_list[n])
    list_of_areas.insert(0, white_area)
    list_of_areas.insert(1, yellow_area)
    list_of_areas.insert(2, red_area)
    tuple_area = tuple(list_of_areas)
    return tuple_area

Answer 1

通过使用lru_缓存，可以使函数在需要时间的python代码中变得高效，它存储缓存并减少运行时间

语法-

# importing module
from functools import lru_cache

# we use lru_cache as a decorator , it takes maxsize argument
@lru_cache(maxsize=10)
def exampleFunction(str):
    time.sleep(2s)
    print(str)
# exampleFunction will take 2 seconds here to run -
exampleFunction ("when function called first time i took 2 seconds to run")

# now exampleFunction will not take a second to run as it is stored in cache
exampleFunction("exampleFunction didn't take a single second now to run")