第八题 剑指 Offer 55 - I. 二叉树的深度

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

题目描述

输入一棵二叉树的根节点，求该树的深度。从根节点到叶节点依次经过的节点（含根、叶节点）形成树的一条路径，最长路径的长度为树的深度。

例如：

给定二叉树 [3,9,20,null,null,15,7]，

返回它的最大深度 3 。

提示

节点总数 <= 10000

思路

这周开始的话主要是学BFS，用DFS解题更快，更简单，且易理解，但这题还是使用BFS方法进行解题

BFS

• 根据二叉树定义数据结构：

let root = {
    val: 3,
    left:{
        val: 9,
        left: null,
        right: null
    },
    right:{
        val: 20,
        left: {
            val: 15
        },
        right: {
            val: 7
        }
    }
}

• 逐层遍历访问树
• 设置 数组temp 保存当前节点的左右子节点
• 访问当前节点的所有子节点：
  left 存在：将当前节点的左子节点 push进 temp
  righ 存在：将当前节点的右子节点 push进 temp

完整代码

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
//二叉树 数据结构
let root = {
    val: 3,
    left:{
        val: 9,
    },
    right:{
        val: 20,
        left: {
            val: 15,
            left: null,
            right: null
        },
        right: {
            val: 7,
            left: null,
            right: null
        }
    }
}

var maxDepth = function(root) {
    // 判空
   if(!root){
       return 0;
   }
   // 每层遍历的数组 bfs
   let bfs = [root];
   // 树的深度
   let depth = 0;
   while(bfs.length){
       // 用于保存当前节点的 左右子节点
       let temp = [];
       // 深度 +1
       depth ++;
       // 遍历当前层的所有节点
       for(let i=0; i<bfs.length; i++){
           if(bfs[i].left){
               temp.push(bfs[i].left);
           }
           if(bfs[i].right){
               temp.push(bfs[i].right);
           }
       }
       // 更新为下一层的节点
       bfs = temp;
   }
   return depth;

};