面试题 04.12. 求和路径

题目

给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。

示例：

给定如下二叉树，以及目标和 sum = 22，
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

分析： 二叉树的路径和问题，由于起点可以是任意节点，因此需要对每一个节点进行遍历，判断以当前节点为路径起始节点的二叉子树是否存在路径满足条件。时间复杂度为$O(n^2)$

class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> int:
        total_path = []
        #DFS判断是否存在路径和满足条件
        def DLR(root, s):
            if s==0:
                total_path.append(path)
            if root == None:
                return
            left = root.left
            right = root.right
            if left != None:
                DLR(left, s-left.val)
            if right != None:
                DLR(right, s-right.val)
            return
        if not root:
            return 0
        #二叉树的循环遍历
        queue = [root]
        while queue:
            root = queue.pop(0)
            if root.left:
                queue.append(root.left)
            if root.right:
                queue.append(root.right)
            DLR(root, sum-root.val)
        return len(total_path)