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

时间:2022-12-09 04:32:31

题目描述

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

题目地址

https://www.nowcoder.com/practice/b736e784e3e34731af99065031301bca?tpId=13&tqId=11177&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

思路

递归,深度优先搜索,使用前序遍历。

每次遍历,判断当前root是否同时满足:

与给定数值相等

左子树为空

右子树为空

如果满足条件,返回val值

如果不是叶子结点,我们分别对根结点的左子树右子树进行递归,直到找到叶子结点,然后遍历把叶子结点和父节点对应的val组成的序列返回上一层;如果叶子结点不满足与给定值相等,其实也返回了序列,只不过是空序列。

Python

# -*- coding:utf-8 -*-
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
node1 = TreeNode(8)
node2 = TreeNode(6)
node3 = TreeNode(10)
node4 = TreeNode(5)
node5 = TreeNode(7)
node6 = TreeNode(9)
node7 = TreeNode(1)
node1.left = node2
node1.right = node3
node2.left = node4
node2.right = node5
node3.left = node6
node3.right = node7 class Solution:
# 返回二维列表,内部每个列表表示找到的路径
def FindPath(self, root, expectNumber):
# write code here
result = []
if not root:
return result
# 如果找到叶子结点,则返回
if not root.left and not root.right and root.val == expectNumber:
return [[root.val]]
else:
# 如果没有找到,对左右子树分别进行递归
left = self.FindPath(root.left, expectNumber-root.val)
right = self.FindPath(root.right, expectNumber-root.val)
for item in left+right:
result.append([root.val]+item)
return result if __name__ == '__main__':
result = Solution().FindPath(node1, 19)
print(result)