二叉树的BFS和DFS详细python代码

最新推荐文章于 2023-12-24 22:27:31 发布

原创

最新推荐文章于 2023-12-24 22:27:31 发布 · 944 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #二叉树 #数据结构 #dfs

本文详细介绍了二叉树的三种深度优先搜索（DFS）——前序遍历、中序遍历、后序遍历，以及层次遍历（BFS）的Python代码实现，包括递归和迭代两种方法。

二叉树的三种深度遍历和层次遍历详细代码

一、DFS

1.二叉树的前序遍历

思路：迭代维护一个堆栈来遍历

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def preorderTraversal(self, root: TreeNode) -> List[int]:
        stack=[]
        res=[]
        if not root:
            return []
        node=root
        while stack or node:
            while node:
                res.append(node.val)
                stack.append(node)#将这个节点的左子树都加入到堆栈里面
                node=node.left
            #如果到了最后一个左子树，即一个节点没有左子树，那就取他的右子树
            node=stack.pop()#q从栈里面取出没有左子树的节点，往回
            node=node.right#来得到右子树
        return res

思路：迭代

##从根节点开始，每次迭代弹出当前栈顶元素，并将其孩子节点压入栈中，先压右孩子再压左孩子。
#在这个算法中，输出到最终结果的顺序按照 Top->Bottom 和 Left->Right，符合前序遍历的顺序。
#这些算法多半也是需要记忆的巴
class Solution(object):
    def preorderTraversal(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        if root is None:
            return []
        stack, output = [root, ], []
        while stack:
            root = stack