leetcode102. 二叉树的层次遍历 &111&104&107

最新推荐文章于 2022-02-03 15:11:15 发布

fairy净

最新推荐文章于 2022-02-03 15:11:15 发布

阅读量239

点赞数

CC 4.0 BY-SA版权

分类专栏：算法-leetcode 二叉树

本文链接：https://blog.youkuaiyun.com/u014204761/article/details/82935790

算法-leetcode 同时被 2 个专栏收录

50 篇文章

订阅专栏

二叉树

7 篇文章

订阅专栏

本文详细介绍了二叉树的层次遍历算法，包括普通层次遍历、逆序层次遍历、最大深度和最小深度的求解。通过具体代码实现，展示了如何使用栈和队列来高效地进行层次遍历。

def levelOrder(self, root):
        """
        和普通的层次遍历不一样的地方是输出结果是每层一个列表的形式
        """
        if not root:
            return []
        res = []
        arr = []
        stack = [root]  ####存放每一层的子节点
        tmp = []
        while stack: 
            while stack:
                root = stack.pop(0)
                arr.append(root.val)
                if root.left:
                    tmp.append(root.left)
                if root.right:
                    tmp.append(root.right)
            res.append(arr)
            arr = []
            stack = tmp
            tmp = []
        return res

leetcode107 二叉树的层次遍历II只需要把结果逆序即可。return res[::-1]

leetcode 104 二叉树的最大深度

def maxDepth(self, root):
        """
        最大深度也是层次遍历
        """
        if not root:
            return 0
        count = 0
        arr = []
        stack = [root]  ####存放每一层的子节点
        tmp = []
        while stack: 
            while stack:
                root = stack.pop(0)
                arr.append(root.val)
                if root.left:
                    tmp.append(root.left)
                if root.right:
                    tmp.append(root.right)
            count += 1
            arr = []
            stack = tmp
            tmp = []
        return count

leetcode 111 二叉树的最小深度

def minDepth(self, root):
        """
        最小深度在层次遍历上加个判定条件即可
        """
        if not root:
            return 0
        count = 0
        arr = []
        stack = [root]  ####存放每一层的子节点
        tmp = []
        while stack: 
            while stack:
                root = stack.pop(0)
                arr.append(root.val)
                if root.left:
                    tmp.append(root.left)
                if root.right:
                    tmp.append(root.right)
                ###如果出现叶子节点 返回
                if not root.left and not root.right:
                    count += 1
                    return count
            count += 1
            arr = []
            stack = tmp
            tmp = []
        return count