Python實現桿切割算法

Question

我正在練習動態編程，也是python的新手。 當我在shell中嘗試這個。

def cut_rod(p, n):
    if n==0:
        return 0
    q = float('-inf')
    for i in range(n):
        q = max(q, p[i] + cut_rod(p, n-1-i))
    return q

p = [1,5,8,9,10,17,17,20,24,30]
print(timeit.repeat("cut_rod(p, 4)", "from __main__ import cut_rod",
              number =1000))

它警告我，p不是一個全局變量。 在timeit函數中p不是已經是全局變量嗎？

Answer 1

問題是你沒有導入p ：

print(timeit.repeat("cut_rod(p, 4)", "from __main__ import cut_rod, p",
              number =1000))

timeit在受控環境中執行代碼，因此您必須顯式傳遞語句所需的所有內容。