剑指offer 面试34题

面试34题:

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

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

解题代码:

# -*- 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
        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

猜你喜欢

转载自www.cnblogs.com/yanmk/p/9219920.html