[LeetCode] Maximum Depth of N-ary Tree N叉树的最大深度

 

Given a n-ary tree, find its maximum depth.

The maximum depth is the number of nodes along the longest path from the root node down to the farthest leaf node.

For example, given a 3-ary tree:

 

 

We should return its max depth, which is 3.

 

Note:

1. The depth of the tree is at most 1000.
2. The total number of nodes is at most 5000.

 

这道题让咱们求一个N叉树的最大深度，因为以前作过 Maximum Depth of Binary Tree 那道题，因此有了求二叉树的基础后，求N叉树也就不难了，无非就是稍稍变换了一下吗，由固定的左右子结点变成了一个一堆子结点，可是方法仍是没有变。首先来看一种常见的递归解法，就是须要有一个当前深度，而后带一个全局变量res进去。在递归函数中，若是node为空，直接返回。若子结点数组为空，那么结果res和cur比较取较大值。不然就遍历子结点数组，对每一个子结点调用递归函数，这里的cur要自增1，参见代码以下：

 

解法一：

class Solution {
public:
    int maxDepth(Node* root) {
        int res = 0;
        helper(root, 1, res);
        return res;
    }
    void helper(Node* node, int cur, int& res) {
        if (!node) return;
        if (node->children.empty()) res = max(res, cur);
        for (Node* child : node->children) {
            helper(child, cur + 1, res);
        }
    }
};

 

咱们也能够不使用其余的函数，直接主函数中递归，首先判空，不然就是遍历子结点数组，而后对每一个子结点调用递归函数的返回值加1后跟res相比，取较大值更新结果res，参见代码以下：

 

解法二：

class Solution {
public:
    int maxDepth(Node* root) {
        if (!root) return 0;
        int res = 1;
        for (Node* child : root->children) {
            res = max(res, maxDepth(child) + 1);
        }
        return res;
    }
};

 

咱们也能够不使用递归，而是用迭代的形式，这里借助队列queue来作，就是BFS的经典写法，不算难，参见代码以下：

 

解法三：

class Solution {
public:
    int maxDepth(Node* root) {
        if (!root) return 0;
        int res = 0;
        queue<Node*> q{{root}};
        while (!q.empty()) {
            for (int i = q.size(); i > 0; --i) {
                auto t = q.front(); q.pop();
                for (auto child : t->children) {
                    if (child) q.push(child);
                }
            }
            ++res;
        }
        return res;
    }
}; 

 

相似题目：

Maximum Depth of Binary Tree

 

参考资料：

https://leetcode.com/problems/maximum-depth-of-n-ary-tree/

https://leetcode.com/problems/maximum-depth-of-n-ary-tree/discuss/148544/Java-Top-down-DFS-solutions

https://leetcode.com/problems/maximum-depth-of-n-ary-tree/discuss/167010/DFS-and-BFS-solutions-in-C%2B%2B.

https://leetcode.com/problems/maximum-depth-of-n-ary-tree/discuss/151804/Solution-Python-C%2B%2B-Simple-with-explanation

 

LeetCode All in One 题目讲解汇总(持续更新中...)