一则算法题引出的关于Python函数参数和变量的思考

在逛segmentfault的时候,看到一个比较有意思的算法题:python怎么获得二叉树根到所有叶子的路径?

然后题主贴出了自己的代码,大概就是这样子的

class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None
def dfs(node, result, tmp):
    if node == None:
        return 
    tmp.append(node)
    if node.left == None and node.right == None:
        result.append([i.val for i in tmp])
        return
    dfs(node.left, result, tmp)
    dfs(node.right, result, tmp)

问题提得还是很规矩,该告知的信息都告知了,所以忍不住想回答一下。

初看这段代码,不知道大家是否能立刻察觉到这段代码有一些很值得注意的地方。嗯,就是他在函数传参的时候使用了列表。在初学Python的时候,无论是书上还是大神的博客上,都告诉我们,不要使用可变参数作为函数参数,否则参数会保持上一次被改变的状态。现在回过头来看看上面给出的代码,tmpresult在每次递归调用后,它们的状态都会被改变。result的改变是我们可以预见并且也是期望的,但是tmp却不好控制。如果我们在遍历左子树的时候把tmp当成参数传到下一次的递归调用中,然后当我们遍历右子树的时候,其添加了左子树的状态还在,所以会出现『tmp数组会保留之前遍历完左子树的状态』这个问题,知道了问题,那么剩下的就是找方法解决这个问题了。

由于左子树和右子树遍历不能互相影响,所以光用一个tmp是不行的,需要一份它的拷贝,注意,这里的拷贝如果只是简单的tmp1 = tmp赋值的话,你会发现效果还是和前面的一样,因为tmp1tmp指向了同一块地址。对于tmp1和tmp任意一个的修改都会影响对方。正确的做法是使用copy模块的copy或者deepcopy,也就是常说的浅拷贝和深拷贝。两者也是有一些区别的,浅拷贝会把原来的内容都做一份拷贝,如果原来的内容中有可变对象,它会持有该对象的引用,无论是原来的变量或者拷贝的对象对可变对象的改动都会影响到另外一个,而深拷贝不会存在这个问题,它会把所有内容的值都拷贝一份,包括可变对象。比如

>>>a = [1, 2, [3, 4]]
>>>import copy
>>>b = copy.copy(a)
>>>b
[1, 2, [3, 4]]
>>> b[1] = 5
>>>b
[1, 5, [3, 4]]  # 修改后的b
>>> a
[1, 2, [3, 4]]  # 改变不可变对象,那么原对象不受影响
>>>b[2].append(5)
>>>b
[1, 5, [3, 4, 5]] # 当前b的值
>>>a
[1, 2, [3, 4, 5]]  # a 被b的操作影响了,浅拷贝『共享可变对象』

而如果是deepcopy,可以看看同样的操作产生的结果

>>>a = [1, 2, [3, 4]]
>>>b = copy.deepcopy(a)
>>>b
[1, 2, [3, 4]]
>>>b[2].append(5)
>>>b
[1, 2, [3, 4, 5]]  # b当前的值
>>>a
[1, 2, [3, 4]]    # a当前的值并不会受到影响,深拷贝不会共享可变对象

最后贴上改正过后的代码,『遍历获取二叉树的所有可达路径』

import copy


class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None


def dfs(node, result, tmp=list()):
    if node is None:
        return

    tmp.append(node)
    tmp1 = copy.copy(tmp)

    if node.left is None and node.right is None:
        result.append([i.val for i in tmp])
        return

    if node.left is not None:
        dfs(node.left, result, tmp)
    if node.right is not None:
        dfs(node.right, result, tmp1)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 1. Java基础部分 基础部分的顺序:基本语法,类相关的语法,内部类的语法,继承相关的语法,异常的语法,线程的语...
    子非鱼_t_阅读 32,602评论 18 399
  • *面试心声:其实这些题本人都没怎么背,但是在上海 两周半 面了大约10家 收到差不多3个offer,总结起来就是把...
    Dove_iOS阅读 27,395评论 30 472
  • 今天是受伤第五天,感觉疼痛感又减轻了一些。但是久站和久坐都不舒服,躺着也不舒服,一天就以三种姿势轮换着。 婆婆今天...
    心际流尘阅读 1,121评论 0 0
  • 1.从本篇文章中我学到的最重要的概念 学习英语是一个日积月累的过程,并且最重要的是怀着兴趣。 同时,语言是用来交流...
    郭媛1号阅读 1,841评论 2 0
  • 三肱二头 肱三头 杠铃弯举5*4。杠铃仰卧伸展5*4 上斜哑铃弯举 5*4滑轮下拉50*4 斜托弯举。5*4 ...
    中山市古镇高级中学方晶阅读 3,957评论 1 0

友情链接更多精彩内容