要求:
从树的根节点开始往下访问一直到叶子结点经过的所有结点形成一条路径。找到所有的这些路径,使其满足这条路径上所有节点数据的和等于给定的整数。
分析:
可以通过对二叉树的遍历找出所有的路径,然后判断各条路径上的所有结点的值的和是否满足给定的整数相等,如果相等,则打印出这条路径。实现思路为:对二叉树进行先序遍历,把遍历路径记录下来,当遍历到叶结点时,判断当前路径上所有结点数据之和是否等于给定整数,如果相等则输出路径信息。
实现代码:
# -*- 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])