剑指Offer(Python多种思路实现):二叉树中和为某一值的路径

剑指Offer(Python多种思路实现):二叉树中和为某一值的路径

面试34题:

题目:二叉树中和为某一值的路径

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

解题思路:

class Solution:
    def FindPath(self, root, expectNumber):
        # write code here
        if not root:
            return []
        result=[]
        
        def FindPathCore(root,path,currentNum):
            currentNum += root.val
            path.append(root)
            # 判断是否达到叶子节点
            flag = (root.left==None and root.right==None)
            
            #如果到达叶子节点且当前值等于期望值
            if currentNum==expectNumber and flag:
                onepath=[]
                for node in path:
                    onepath.append(node.val)
                result.append(onepath)
                
            if currentNum<expectNumber:
                if root.left:
                    FindPathCore(root.left,path,currentNum)
                if root.right:
                    FindPathCore(root.right,path,currentNum)
            path.pop()
                
        FindPathCore(root,[],0)
        return result
发布了44 篇原创文章 · 获赞 2 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_44151089/article/details/104471386