【Leetcode】python回溯算法结果出现[[], [], []](列表深拷贝 浅拷贝问题)

问题

Leetcode 77.组合
代码如下:

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        ans = []
        def search(start, end, path, k):
            if len(path) == k:
                ans.append(path)
                print(path)
                return
            
            for i in range(start, end + 1):
                path.append(i)
                search(i + 1, end, path, k)
                path.pop()

        search(1, n, [], k)
        # print(ans)
        return ans

结果如图:路径是正确的,但为什么答案中全是[]?
【Leetcode】python回溯算法结果出现[[], [], []](列表深拷贝 浅拷贝问题)_第1张图片

解释

由于我们append的是path,是直接赋值。相当于把同一个path在ans中链接了n次,原始的path改变会使得ans中的path跟着改变。也就是说,虽然我们最后搜索到了底部,但我们回溯到头的过程中又把原来的列表变回空了。所以最后答案中返回的是空。

解决方案

path.copy()path[:]来替代path

注意

如果用path.copy()是浅拷贝,只在path是一维的时候正确。如果path是多维,如:path = [1, 2, [3, 4]],那么path.copy()只是对最外层做了拷贝,里面的[3, 4]并没有被复制。也就是说此时如果我在[3, 4]中添加一个数字变成[3, 4, 5],那么即便copy过,依旧会改变它的值。此时如果希望append出来的path完全独立,需要使用深拷贝:path.deepcopy()
参考链接:https://blog.csdn.net/nini_k/article/details/123712582

你可能感兴趣的:(python,算法,leetcode)