一则算法题引出的关于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)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,634评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,951评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,427评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,770评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,835评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,799评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,768评论 3 416
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,544评论 0 271
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,979评论 1 308
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,271评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,427评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,121评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,756评论 3 324
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,375评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,579评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,410评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,315评论 2 352

推荐阅读更多精彩内容

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