104. 二叉树的最大深度 JavaScript实现

104. 二叉树的最大深度

题目链接
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

一、层次遍历

对二叉树进行层次遍历，用一个变量指控层数即可。

代码实现

var maxDepth = function(root) {
    if(!root) return 0;
    let queue = [];

    // 初始化队列
    queue.push(root);

    // 控制层数
    let level = 0;

    // 遍历二叉树,队列不为空
    while(queue.length){
        // 弹出的数量是当前队列中的数量
        const n = queue.length;

        // 控制每次弹出的数量，数量用队列里的节点数量控制比较好。
        // 如果用Math.power(2,level),会出现弹出为空的情况
        for(let i=0;i<n;i++){
            const node = queue.shift();

            // 并且去检查弹出的节点是否存在左右节点
            if(node.left) queue.push(node.left);
            if(node.right) queue.push(node.right);
        }
        level++;    
    }

    return level;
};

二、深度优先遍历

思想和最小深度一样，递归的找到左右子树的最大深度，然后加上自身即可。注意最小深度需要考虑单侧节点为空的情况。

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */

var maxDepth = function(root) {
    if(!root) return 0

    let left = maxDepth(root.left)
    let right = maxDepth(root.right)

    return Math.max(left,right)+1
};