python3实现在二叉树中找出和为某一值的所有路径

在二叉树中找出和为某一值的所有路径
请写一个程序创建一棵二叉树,并按照一定规则,输出二叉树根节点到叶子节点的路径。
规则如下:
1、从最顶端的根结点,到最下面的叶子节点,计算路径通过的所有节点的和,如果与设置的某一值的相同,那么输出这条路径上的所有节点。
2、从根节点遍历树时,请请按照左到右遍历,即优先访问左子树的节点。
二叉树创建规则:从上到下一层一层的,按照从左到右的顺序进行构造
输入"10,5,12,4,7"值,构造的树如下:
1) 10
2) 10
      /
    5

3) 10
       /\
     5 12
4) 10
        /\
      5 12
     /
   4

5) 10
        /\
      5 12
      /\
     4 7
针对上面的二叉树,如果当前我们设置的“路径和”为19,那么输出结果为:
10,5,4
如果有多个路径,按到左到右的顺序遍历生成的结果每行显示一个显示。例如如果当前我们设置的“路径和”为22,那么输出结果为:
10,5,7
10,12
如果没有找到路径和为设置的值的路径,输出error。
三、输入:
输入整数N---路径和
一行字符串,多个正整数,之间用","隔开
四、输出: 满足条件的二叉树路径
五、样例输入:
22
10,5,12,4,7

六、样例输出:
10,5,7
10,12

demo:

class Node(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Tree(object):
    lt = []  # 依次存放左右孩子未满的节点

    def __init__(self):
        self.root = None

    def add(self, number):
        node = Node(number)  # 将输入的数字节点化,使其具有左右孩子的属性
        if self.root == None:
            self.root = node
            Tree.lt.append(self.root)
        else:
            while True:
                point = Tree.lt[0] # 依次对左右孩子未满的节点分配孩子
                if point.left ==None:
                    point.left = node
                    Tree.lt.append(point.left)  # 该节点后面作为父节点也是未满的,也要加入到列表中。
                    return
                elif point.right ==None:
                    point.right = node
                    Tree.lt.append(point.right)  # 与左孩子同理
                    Tree.lt.pop(0)  # 表示该节点已拥有左右孩子,从未满列表中去除
                    return

class Solution:
    def __init__(self):
        self.results = []
    def RecursionFindPath(self, root, expectNumber, result):
        result.append(root.val)
        if root.left == None and root.right == None and sum(result) == expectNumber:
            self.results.append(result)
        temp = result[:]
        if root.left:
            self.RecursionFindPath(root.left, expectNumber, result)
        result = temp[:]
        if root.right:
            self.RecursionFindPath(root.right, expectNumber, result)
    def FindPath(self, root, expectNumber):
        if root == None:
            return []
        self.RecursionFindPath(root, expectNumber, [])
        self.results = sorted(self.results, key=len, reverse=True)
        return self.results


if __name__ =='__main__':
    t = Tree()  # 二叉树类的实例化
    L = [10, 5, 12, 4, 7]
    for i in L:
        t.add(i)

    expectNum = 22
    print(Solution().FindPath(t.root, expectNum))

输出样例:

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值