剑指offer第二版(Python3)--面试题34:二叉树中和为某一值的路径

第2章 面试需要的基础知识

第3章 高质量的代码

第4章 解决面试题的思路

  面试题27 :二叉树的镜像

  面试题29 :顺时针打印矩阵

  面试题30 :包含min函数的栈

  面试题31:栈的压入、弹出序列

  面试题32:上往下打印二叉树

  面试题33: 二叉搜索树的后序遍历序列

  面试题34: 二叉树中和为某一值的路径

第5章 优化时间和空间效率

第6章 面试中的各项能力

第7章 两个面试案例


题目描述
牛客网

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

解题思路

牛客网

  1. 首先该题是基于递归去遍历整棵树,遍历完每一条路径,遍历的顺序是先根节点,然后是左节点,接着是右节点,遍历过程中如果累加和大于给定数值,需停止往下遍历,也就是剪枝;
  2. 如果节点的左右子树都为空,且路径之和等于参数,就说明该路径是需要输出的。

实战

class Solution:
    # 返回二维列表,内部每个列表表示找到的路径
    def FindPath(self, root, expectNumber):
        # write code here
        def core(node, sums, lst):
            if sums+node.val == expectNumber and not node.left and not node.right:
                results.append(lst+[node.val])
            if sums > expectNumber:
                return
            
            if node.left:
                core(node.left, sums+node.val, lst+[node.val])
            if node.right:
                core(node.right, sums+node.val, lst+[node.val])
        
        if not root:
            return []
        results = []
        core(root, 0, [])
        return sorted(results, key=lambda x: len(x), reverse=True)

你可能感兴趣的:(算法设计,剑指offer,二叉树中和为某一值得路径)