24.二叉树中和为某一值的路径(python)

题目描述

输入一颗二叉树的跟节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)

思路： 层次遍历，每次记录当前路径，最后遍历到叶子，满足条件则insert,每次insert到第一个位置，所以前面自然是最长的。

import copy
class Solution:
    # 返回二维列表，内部每个列表表示找到的路径
    def FindPath(self, root, expectNumber):
        # write code here
        if root == None:
            return []
        support = [root]
        supportArrayList = [[root.val]]
        ret=[]
        while support:
            tmpNode = support[0]
            tmpArrayList=supportArrayList[0]
            if tmpNode.left==None and tmpNode.right == None:
                if sum(tmpArrayList)==expectNumber:
                    ret.insert(0,tmpArrayList)
            if tmpNode.left:
                support.append(tmpNode.left)
                newtmpArrayList =copy.copy(tmpArrayList)
                newtmpArrayList.append(tmpNode.left.val)
                supportArrayList.append(newtmpArrayList)
            if tmpNode.right:
                support.append(tmpNode.right)
                newtmpArrayList = copy.copy(tmpArrayList)
                newtmpArrayList.append(tmpNode.right.val)
                supportArrayList.append(newtmpArrayList)
            del support[0]
            del supportArrayList[0]
        return ret

2019-12-15 09:28:59