Python算法——在二叉树中找出与输入整数相等的所有路径

最新推荐文章于 2020-08-28 11:03:51 发布

Aamax

最新推荐文章于 2020-08-28 11:03:51 发布

阅读量296

点赞数 1

CC 4.0 BY-SA版权

分类专栏：算法

本文链接：https://blog.youkuaiyun.com/xjtuse123/article/details/98735330

该博客介绍了如何在二叉树中寻找所有节点值之和等于给定整数的路径。通过先序遍历二叉树，记录路径，并在到达叶节点时检查路径和，若和等于目标值，则输出路径。实现代码的时间复杂度为O(N)，与二叉树先序遍历相同。

要求：

从树的根节点开始往下访问一直到叶子结点经过的所有结点形成一条路径。找到所有的这些路径，使其满足这条路径上所有节点数据的和等于给定的整数。

分析：

可以通过对二叉树的遍历找出所有的路径，然后判断各条路径上的所有结点的值的和是否满足给定的整数相等，如果相等，则打印出这条路径。实现思路为：对二叉树进行先序遍历，把遍历路径记录下来，当遍历到叶结点时，判断当前路径上所有结点数据之和是否等于给定整数，如果相等则输出路径信息。

实现代码：

# -*- coding:utf-8 -*-
class BiTNode():
    def __init__(self):
        self.data = None
        self.lchild = None
        self.rchild = None

'''
方法功能：打印满足所有结点数据和等于num的所有路径
参数：root--二叉树根节点 num--给定整数  sum--当前路径上所有结点和  v--根结点到当前遍历结点路径
'''
def FindRoad(root,num,sums,v):
    #记录当前遍历root结点
    sums += root.data
    v.append(root.data)
    #当前结点是叶子结点且遍历路径上所有结点和等于num
    if root.lchild == None and root.rchild == None and sums == num:
        i = 0
        while i < len(v):
            print(v[i])
            i += 1
    #遍历root左子树
    if root.lchild != None:
        FindRoad(root.lchild,num,sums,v)
    #遍历root右子树
    if root.rchi