【剑指 Offer 55 - I. 二叉树的深度】

【剑指 Offer 55 - I. 二叉树的深度】

题目描述:

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-shu-de-shen-du-lcof/

  • 输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
  • 说明: 叶子节点是指没有子节点的节点。

示例:

  • 示例1
  • 给定二叉树 [3,9,20,null,null,15,7],

			    3
			   / \
			  9  20
			    /  \
			   15   7

     - 返回它的最大深度 3 。

  • 示例2

输入: [1, 2, 3, 2, 2, 2, 5, 4, 2]
输出: 2

  • 提示
     1 <= 数组长度 <= 50000

前提知识

  • 树的遍历方式总体分为两类:深度优先搜索(DFS)广度优先搜索(BFS)
    常见的 DFS : 先序遍历(中左右)、中序遍历(左中右)、后序遍历(左右中)
    常见的 BFS : 层序遍历(即按层遍历)

解析思路1:后序遍历(DFS)->递归 :max(左、右子树层数)+1

  • 算法流程示意图
    【剑指 Offer 55 - I. 二叉树的深度】
    【剑指 Offer 55 - I. 二叉树的深度】
    【剑指 Offer 55 - I. 二叉树的深度】- 同理:
    【剑指 Offer 55 - I. 二叉树的深度】

代码(cpp)

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {

public:
    int maxDepth(TreeNode* root) {

        if (root == NULL)  return 0;
        int leftDepth = maxDepth(root->left);  //左子树深度
        int rightDepth = maxDepth(root->right);  //右子树深
        return max(leftDepth, rightDepth)+1  ;  //当前节点的深度
    }
};

代码(python3)


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

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        if not root: return 0
        return max(self.maxDepth(root.left), self.maxDepth(root.right)) + 1

复杂度分析:

  • 时间复杂度 O(N) : N 为树的节点数量,计算树的深度需要遍历所有节点。
  • 空间复杂度 O(N) : 最差情况下(当树退化为链表时),递归深度可达到 N 。

解析思路2:层序遍历(BFS)

  • 参考k神-jdy-解析.

  • 广度优先搜索(BFS)

  • 树的层序遍历 / 广度优先搜索往往利用 队列 实现。

  • 核心问题: 每遍历一层,则计数器 +1 ,直到遍历完成,则可得到树的深度。

  • 算法流程

【剑指 Offer 55 - I. 二叉树的深度】
【剑指 Offer 55 - I. 二叉树的深度】
【剑指 Offer 55 - I. 二叉树的深度】
【剑指 Offer 55 - I. 二叉树的深度】
【剑指 Offer 55 - I. 二叉树的深度】
【剑指 Offer 55 - I. 二叉树的深度】

代码(cpp)

注意 遍历每层时候,利用队列存放、遍历删除当前层节点的代码过程


/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
// private:
//     int res = 0;
public:
    int maxDepth(TreeNode* root) {
       if (root ==NULL) return 0;
       queue<TreeNode*> q;   //创建空队列q
       q.push(root); 	 	 //将root推入队列q
       int res = 0;
       while(!q.empty()){
           ++ res;
           for(int i=q.size();i>0;i--){   //遍历队列q中的当前层节点
            //    TreeNode* node = q.front();
               auto node = q.front(); 
               q.pop(); // 删除队列中当前层节点
               if (node->left ) q.push(node->left);
               if (node->right) q.push(node->right);
           }
       }
        return res;
    }
};

代码(python3)

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        if not root: return 0
        queue, res = [root], 0
        while queue:
            tmp = []
            for node in queue:
                if node.left: tmp.append(node.left)
                if node.right: tmp.append(node.right)
            queue = tmp
            res += 1
        return res

复杂度分析:

  • 时间复杂度 O(N) : N 为树的节点数量,计算树的深度需要遍历所有节点。
  • 空间复杂度 O(N) : 最差情况下**(当树平衡时)**,队列 queue 同时存储 N/2 个节点
上一篇:【JAVASE小新】博客练习声明异常


下一篇:阿里云资深专家李国强:云原生的一些趋势和新方向