python求二叉树最大深度

最新推荐文章于 2024-03-14 09:39:09 发布

Anut_hey

最新推荐文章于 2024-03-14 09:39:09 发布

阅读量1.7k

点赞数

分类专栏：算法文章标签： python leetcode 算法

本文链接：https://blog.csdn.net/Anut_hey/article/details/120556764

版权

算法专栏收录该内容

20 篇文章 0 订阅

订阅专栏

这篇博客介绍了计算二叉树最大深度的三种方法：递归、BFS（广度优先搜索）和DFS（深度优先搜索）。递归方法通过比较左右子树的深度找到最大值；BFS使用队列逐层遍历；DFS则利用双栈保存节点和深度信息。每种方法都有其独特的优势和应用场景。

摘要由CSDN通过智能技术生成

三种方法，作记录用
1递归：在每个节点选择最大的深度+1返回(左子树或右子树的深度)

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        
        if (root==None):
            return 0
        num_left = self.maxDepth(root.left) + 1
        num_right = self.maxDepth(root.right) + 1

        if num_left >= num_right:
            return num_left
        
        return num_right

2BFS：使用队列，按层遍历(每次迭代一层)

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        if(root==None):
            return 0
        queue = [root]
        count = 0
        while(queue):
            count += 1
            size = len(queue)
            while(size):
                size -= 1
                node = queue.pop(0)
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
        
        return count

3DFS：使用双栈，一个存储节点，一个记录该节点所在深度，并记录下遍历过程中的深度最大值

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        if(root==None):
            return 0
        stack1 = [root]
        stack2 = [1]
        maxd = 1
        while(stack1):

            node = stack1.pop()
            depth = stack2.pop()
            maxd = max(maxd,depth)

            if node.left:
                stack1.append(node.left)
                stack2.append(depth + 1)
            if node.right:
                stack1.append(node.right)
                stack2.append(depth + 1)
        
        return maxd