二叉樹中和為某一值的路徑(python)

題目描述

輸入一顆二叉樹的跟節點和一個整數，打印出二叉樹中結點值的和為輸入整數的所有路徑。路徑定義為從樹的根結點開始往下一直到葉結點所經過的結點形成一條路徑。(注意: 在返回值的list中，數組長度大的數組靠前)

 

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回二維列表，內部每個列表表示找到的路徑
    def FindPath(self, root, expectNumber):
        # write code here
        #首先要理解題意，是從根節點往子節點連。
        #如果只有根節點或者找到葉子節點，我們就把其對應的val值返回
        #如果不是葉子節點，我們分別對根節點的左子樹、右子樹進行遞歸，
        #直到找到葉子結點。然后遍歷把葉子結點和父節點對應的val組成的序列返回上一層；如果沒找到路徑，其實也返回了序列，只不過是[]
        if not root:
            return []
        if root and not root.left and not root.right and root.val==expectNumber:
            return [[root.val]]
        res = []
        left = self.FindPath(root.left,expectNumber-root.val)
        right = self.FindPath(root.right,expectNumber-root.val)
        for i in left+right:
            res.append([root.val]+i)
        return res