[剑指offer] 二叉树中和为某一值的路径

题目内容

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

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

---

题目思路

这是个典型的深度优先搜索题目。我们从根节点到叶节点进行搜索，进行深度（纵向）的构建序列。如果是根节点而且满足和满足，则加入，否则返回上级。如果和已经相等或者超过，却不是叶节点，则同样返回上一级。

---

程序代码

# -*- 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):
        # write code here
        res=[]
        if not root:
            return []
        self.cal(root,expectNumber,res,[root.val])
        return res
    
    def cal(self,root,num,res,tmp):
        if not root.left and not root.right:#是根节点
            if sum(tmp)==num:
                res.append(tmp)
            else:
                return
        elif sum(tmp)>=num:
            return
        if root.left:
            self.cal(root.left,num,res,tmp+[root.left.val])
        if root.right:
            self.cal(root.right,num,res,tmp+[root.right.val])