【算法】求N叉树的最大深度（java）

【算法】求N叉树的最大深度

题目

求N叉树的最大深度

N叉树数的定义

首先我们要明白N叉树的定义是什么，其实就是每个节点下会有N个节点。

用代码定义一个N叉树的一个节点类如下：

class NTreeNode {
    //节点的值
    int val;
    //子节点
    List<NTreeNode> sonTreeNodeList;

    NTreeNode() {}
    NTreeNode(int val, List<NTreeNode> list) {
        this.val = val;
        sonTreeNodeList = list;
    }
}

解体思路

想要求一个给定的N叉树的最大深度。思路如下：

1、遍历N叉树根节点下的【子节点数组】，每次遍历一个节点时，都递归调一次方法

2、递归到最下层的节点时，该节点的【子节点数组】将会是一个空列表，这时递归返回该节点的深度为1，然后继续向上递归返回，并把每个节点对应的深度都添加到一个【保存深度的数组】中

3、最后，返回【保存深度的数组】中的最大元素 + 1

代码实现如下：

public int maxND(NTreeNode root) {
    if (root == null) {
        return 0;
    } else if (root.sonTreeNodeList.isEmpty()) {
        return 1;
    }
    //定义一个【保存深度的数组】
    List<Integer> heights = new LinkedList<>();
    //遍历当前节点下的所有子节点
    for (NTreeNode item : root.sonTreeNodeList) {
        //对每一个节点求最大深度
        int depth = maxND(item);
        //把每个节点对应的深度保存在数组中
        heights.add(depth);
    }
    //返回数组中最大的元素 + 1
    return Collections.max(heights) + 1;
}