题解 | #牛群特殊路径的数量#
牛群特殊路径的数量
https://www.nowcoder.com/practice/1c0f95d8396a432db07945d8fe51a4f5
- 题目考察的知识点 : 二叉树的前序遍历
- 题目解答方法的文字分析:
- 首先定义了一个内部函数,用于前序遍历整棵二叉树,并将所有节点存入一个列表中。在该函数中,我们首先判断当前节是否为空。如果为空,则返回;否则,将当前节点存入列表中,并递归调用左右子节点。
- 其次定义了一个内部函数,用于计算从当前节点出发到各个叶子节点的路径和,并将这些路径和存入一个列表中。在该函数中,我们首先判断当前节点是否为空。如果为空,则返回;否则,如果当前节点是叶子节点,则将从根节点到当前节点的路径和s加入列表中;否则,递归调用左右子节点,并将当前节点值加入s中。
- 最后遍历所有路径和,统计其中等于sum的个数,并将其作为结果返回。
- 本题解析所用的编程语言:Python
- 完整且正确的编程代码
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
#
# @param root TreeNode类
# @param sum int整型
# @return int整型
#
class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> int:
        def preorder(root, array):
            if not root:
                return
            array.append(root)
            preorder(root.left, array)
            preorder(root.right, array)
        
        def path(root, s, array):
            if not root:
                return
            if not root.left and not root.right:
                s += root.val
                array.append(s)
                return
            s += root.val
            array.append(s)
            path(root.left, s, array)
            path(root.right, s, array)
        
        array = []
        result = 0
        preorder(root, array)
        for node in array:
            s_array = []
            path(node, 0, s_array)
            result += s_array.count(sum)
            
        return result
牛客高频top202题解系列 文章被收录于专栏
 记录刷牛客高频202题的解法思路

