LeetCode热题（JS版） - 559. N 叉树的最大深度

题目

给定一个 N 叉树，找到其最大深度。

最大深度是指从根节点到最远叶子节点的最长路径上的节点总数。

N 叉树输入按层序遍历序列化表示，每组子节点由空值分隔（请参见示例）。

示例 1：

输入：root = [1,null,3,2,4,null,5,6]
输出：3

示例 2：

输入：root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出：5

提示：

树的深度不会超过 1000 。
树的节点数目位于 [0, 104] 之间。

思路: 递归

/**
 * Definition for Node.
 * class Node {
 *     val: number
 *     children: Node[]
 *     constructor(val?: number, children?: Node[]) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.children = (children===undefined ? [] : children)
 *     }
 * }
 */

function maxDepth(root: Node | null): number {
    let result = 0

    const DFS = (node, depth) => {
        if(!node) return

        if(node.children) {
            node.children.forEach(child => {
                DFS(child, depth + 1);
            })
        }

        if(depth > result) result = depth;
    }

    DFS(root, 1);

    return result
};

写法上优化下：

    if(root == null) return 0;

    let max = 0;
    for(let child of root.children) {
        max = Math.max(max, maxDepth(child))
    }

    return max + 1