python 二叉树深度优先搜索和广度优先搜索

发布时间:2019-08-05 发布网站:脚本宝典
脚本宝典收集整理的这篇文章主要介绍了python 二叉树深度优先搜索和广度优先搜索脚本宝典觉得挺不错的,现在分享给大家,也给大家做个参考。

class TreeNode:

def __inIT__(self, value=None, left=None, right=None):
    self.value = value
    self.left = left  # 左子树
    self.right = right  # 右子树

@H_304_40@

node1 = TreeNode("A",

             TreeNode("B",
                      TreeNode("D"),
                      TreeNode("E")
                      ),
             TreeNode("C",
                      TreeNode("F"),
                      TreeNode("G")
                      )
             )

def PReTraverse(root):

if root is None:
    return
print(root.value)
preTraverse(root.left)
preTraverse(root.right)

def midTraverse(root):

if root is None:
    return
midTraverse(root.left)
print(root.value)
midTraverse(root.right)

def afterTraverse(root):

if root is None:
    return
afterTraverse(root.left)
afterTraverse(root.right)
print(root.value)

def DFs(root):

res = []
if root is None:
    return res
q = []
q.append(root)
while len(q) > 0:
    r = q.pop()
    print(r.value)
    if r.left is not None:
        # 非空左孩子入队
        q.append(r.left)
    if r.right is not None:
        # 非空右孩子入队
        q.append(r.right)
    res.append(r.value)
return res

def bfs(root):

# write your code here
res = []
# 如果根节点为空,则返回空列表
if root is None:
    return res
# 模拟一个队列储存节点
q = []
# 首先将根节点入队
q.append(root)
# 列表为空时,循环终止
while len(q) > 0:
    length = len(q)
    r = q.pop(0)
    print(r.value)
    if r.left is not None:
        # 非空左孩子入队
        q.append(r.left)
    if r.right is not None:
        # 非空右孩子入队
        q.append(r.right)
    res.append(r.value)
return res

dfs(node1)
print("-------------------")
bfs(node1)

脚本宝典总结

以上是脚本宝典为你收集整理的python 二叉树深度优先搜索和广度优先搜索全部内容,希望文章能够帮你解决python 二叉树深度优先搜索和广度优先搜索所遇到的问题。

如果觉得脚本宝典网站内容还不错,欢迎将脚本宝典推荐好友。

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。