Python：列表乘積

今天在做 leetcode 上一道算法題時遇到一個列表相乘的問題。 題目如下：

給你兩個單詞 word1 和 word2，請你計算出將 word1 轉換成 word2 所使用的最少操作數 。

你可以對一個單詞進行如下三種操作：

• 插入一個字符
• 刪除一個字符
• 替換一個字符

題目鏈接： https://leetcode-cn.com/problems/edit-distance/

 

看完題解后我寫的代碼如下：

class Solution:
    def minDistance(self, word1: str, word2: str) -> int:
        n1, n2 = len(word1), len(word2)

        dp = [[0] * (n2 + 1)] * (n1 + 1)

        dp[0] = list(range(n2 + 1))

        for i in range(n1 + 1):
            dp[i][0] = i

 print(dp)
 for i in range(1, n1 + 1):
            for j in range(1, n2 + 1):
                if word1[i-1] == word2[j-1]:
                    dp[i][j] = dp[i-1][j-1]
                else:
                    dp[i][j] = min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1     # delete, add, modify

        return dp[-1][-1] 

 

但我得到的答案卻一直不對。 我的解法和參考答案只有一行代碼不同，就是上述標紅的代碼。紫色標識的print 代碼打印結果如下：

"""
[
    [0, 1, 2, 3], 
    [5, 0, 0, 0], 
    [5, 0, 0, 0], 
    [5, 0, 0, 0], 
    [5, 0, 0, 0], 
    [5, 0, 0, 0]
]
"""

# 除第一個內嵌列表外，其它的內嵌列表中的第一個元素都是5

 

想了好久才找到問題所在：python列表中的嵌套列表在相乘時，得到的內嵌列表共享同一個內存地址， 如下：

In [24]: lst = [[0]] * 3                                                                                                                                      

In [25]: lst                                                                                                                                                  
Out[25]: [[0], [0], [0]]

In [26]: id(lst[0])                                                                                                                                           
Out[26]: 4579535688

In [27]: id(lst[1])                                                                                                                                           
Out[27]: 4579535688

In [28]: id(lst[0]) == id(lst[1]) and id(lst[1]) == id(lst[2])                                                                                                
Out[28]: True

同理， 上述算法代碼中的內嵌列表也是共享着同一個內存地址，所以才出現了上面的問題。

 

正解解法：

class Solution:
    def minDistance(self, word1: str, word2: str) -> int:
        n1, n2 = len(word1), len(word2)

        dp = [[0 for _ in range(n2 + 1)] for _ in range(n1+1)]  # n1 + 1 每次 for range 的時候，內嵌列表都重新用 list() 實例化一次，所以得到的是不同的列表對象

        dp[0] = list(range(n2 + 1))

        for i in range(n1 + 1):
            dp[i][0] = i

        print(dp)

        for i in range(1, n1 + 1):
            for j in range(1, n2 + 1):
                if word1[i-1] == word2[j-1]:    # dp[i] 表示 word1 的前i個元素，此時word1所在的索引是 i -1 ； j - 1 也是同理
                    dp[i][j] = dp[i-1][j-1]
                else:
                    dp[i][j] = min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1     # delete, add, modify

        return dp[-1][-1]

 

題解參考鏈接：

https://leetcode-cn.com/problems/edit-distance/solution/dong-tai-gui-hua-72-bian-ji-ju-chi-by-bi-okfr/

https://leetcode-cn.com/problems/edit-distance/solution/edit-distance-by-ikaruga/

https://leetcode-cn.com/problems/edit-distance/solution/bian-ji-ju-chi-by-leetcode-solution/