二叉树层次遍历c语言_LeetCode 之二叉树前序、中序、后序、层次遍历

最新推荐文章于 2024-07-28 17:42:24 发布

weixin_39575170

最新推荐文章于 2024-07-28 17:42:24 发布

阅读量267

点赞数

文章标签：二叉树层次遍历c语言二叉树的层次遍历层次遍历二叉树层次遍历求二叉树宽度

1. 前序遍历：

144. 二叉树的前序遍历

前序遍历，输出为[1,2,3,4,5]

递归

def preorderTraversal(self, root):
        res=[]
        def helper(root):
            if not root:
                return 

            res.append(root.val)
            helper(root.left)
            helper(root.right)

        helper(root)
        return res

迭代

 def preorderTraversal(self, root):
        if not root:
            return []
        res = []
        stack = [root]
        while stack:
            node = stack.pop()
            res.append(node.val)
#因为栈的逻辑是先进后出，因此先将右子树放入
            if node.right:
                stack.append(node.right)
            if node.left:
                stack.append(node.left)
        return res

2. 中序遍历

94. 二叉树的中序遍历

中序遍历，输出为[1,2,3,4,5]

递归

def inorderTraversal(self, root):
        res = []
        def helper(root):
            if not root:
                return []
                
            helper(root.left)
            res.append(root.val)
            helper(root.right)

        helper(root)
        return res

迭代

def inorderTraversal(self, root):
        res, stack = [], []
        while stack or root:
            if root:
                stack.append(root)
                root=root.left
            else:
                node = stack.pop()
                res.append(node.val)
                root = node.right
        return res

3. 后序遍历

145. 二叉树的后序遍历

后序遍历，输出为[1,2,3,4,5]

递归

def inorderTraversal(self, root):
        res = []
        def helper(root):
            if not root:
                return []
                
            helper(root.left)
            helper(root.right)
            res.append(root.val)

        helper(root)
        return res

迭代

def preorderTraversal(self, root):
        if not root:
            return []
        res = []
        stack = [root]
        while stack:
            node = stack.pop()
            res.append(node.val)
#此处和前序遍历不同，现将左子树压入栈中
            if node.right:
                stack.append(node.left)
            if node.left:
                stack.append(node.right)
#倒序输出
        return res[::-1]

4. 层次遍历

自上而下层次遍历（广度优先法）

102. 二叉树的层序遍历

def levelOrder(self, root):
        if not root:
            return []
        res = []
        queue = [root]
        while queue:
            size =len(queue)
            temp =[]
            for _ in range(size):
                node = queue.pop(0) 
                temp.append(node.val)
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            res.append(temp)
        return res

自下而上层次遍历

107. 二叉树的层次遍历 II

def levelOrderBottom(self, root):
        if not root:
            return []
        res = []
        queue = [root]
        while queue:
            size =len(queue)
            temp =[]
            for _ in range(size):
                node = queue.pop(0) 
                temp.append(node.val)
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            res.append(temp)
#将自上而下的层次遍历倒序输出
        return res[::-1]