【剑指offer】24.二叉树中和为某一值的路径[by Python]

最新推荐文章于 2020-09-01 21:29:11 发布

Chosen_Xxx

最新推荐文章于 2020-09-01 21:29:11 发布

阅读量130

点赞数

博客围绕二叉树展开，给出题目：输入二叉树和整数，打印结点值和为该整数的所有路径，路径从根结点到叶结点。解题采用深度优先搜索（DFS）方法，先理解题意，再分情况递归处理，最后给出了相应的Python代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

输入一颗二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

解题方法

其实这个就是一个dfs的问题，遇到的比较多了。第一遍没通过是因为忘了把root.val给放进去了！这个可不能忘啊！！

参考Binary Tree Paths

# -*- coding:utf-8 -*-
# -*- 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):
        res = []
        if not root: return res
        self.target = expectNumber
        self.dfs(root, res, [root.val])
        return res

    def dfs(self, root, res, path):
        if not root.left and not root.right and sum(path) == self.target:
            res.append(path)
        if root.left:
            self.dfs(root.left, res, path + [root.left.val])
        if root.right:
            self.dfs(root.right, res, path + [root.right.val])
 
 
 
 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

                版权声明：本文为博主原创文章，未经博主允许不得转载。                    https://blog.youkuaiyun.com/u010636181/article/details/78307595                </div>
                      <link rel="stylesheet" href="https://csdnimg.cn/release/phoenix/template/css/ck_htmledit_views-cd6c485e8b.css">
                          <link rel="stylesheet" href="https://csdnimg.cn/release/phoenix/template/css/ck_htmledit_views-cd6c485e8b.css">
      <div class="htmledit_views" id="content_views">

题目描述

思路：首先要理解题意，是从根节点往子节点连。

1、如果只有根节点或者找到叶子节点，我们就把其对应的val值返回

2、如果不是叶子节点，我们分别对根节点的左子树、右子树进行递归，直到找到叶子结点。然后遍历把叶子结点和父节点对应的val组成的序列返回上一层；如果没找到路径，其实也返回了序列，只不过是[]

代码如下：

# -*- coding:utf-8 -*-
class TreeNode():
	def __init__(self,x):
		self.val = x
		self.left = None
		self.right = None

def function(root,target_number):
result = []
if not root:
return result
# 如果只有根节点或者找到叶子节点，我们就把其值返回
if not root.left and not root.right and root.val == target_number:
return [[root.val]]
else:
# 如果不是叶子节点，我们分别对根节点的左子树、右子树进行递归，注意修改变量:
left = function(root.left,target_number - root.val)
right = function(root.right,target_number - root.val)
for item in left+right：
result.append([root.val]+item)
return result