算法进行时——104. 二叉树的最大深度

Posted 2021-04-09 切图仔军备营

给定一个二叉树，找出其最大深度。

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

说明: 叶子节点是指没有子节点的节点。

示例：给定二叉树 [3,9,null,null,15,7,3,3,43]，

返回它的最大深度 5 。

解题：

//方法一：
/**
 * 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 === null) return 0;
    let depth = 1;
    let stack = [root];
    let currentNode = root;
    
    while(stack.length) {
        while(getNotArrivedChild(currentNode)) {
            stack.push(getNotArrivedChild(currentNode));
            currentNode = getNotArrivedChild(currentNode);
            //这里给访问过的节点标记一下
            currentNode.ok = true;
            //如果加入这个节点后，stack长度大于当前depth，就更新depth为stack长度。
            if(stack.length > depth) depth = stack.length;
        }
        stack.pop();
        currentNode = stack[stack.length-1];
    }
    
    return depth;
};
//返回当前节点还未被访问过的子节点
function getNotArrivedChild(root) {
    if(root.left && !root.left.ok) return root.left;
    if(root.right && !root.right.ok) return root.right;
    return null;
}

//方法二
/**
 * 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) {
    return root == null ? 0 : Math.max(maxDepth(root.left), maxDepth(root.right)) + 1;
}