leetcode#104. 二叉树的最大深度 & leetcode#559. N叉树的最大深度

二叉树的最大深度

题目描述：

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

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

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

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

返回它的最大深度 3 。

N叉树的最大深度

题目描述：

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

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

例如，给定一个 3叉树 :

我们应返回其最大深度，3。

解题思路

  两道题思路类似，均可用几行递归解决问题。
  二叉树采用DFS和后序遍历框架，当前结点为空时，递归结束。核心在于对当前结点的处理，求出当前结点的左右子树的高度，然后取出最大值并+1，返回即可。注意：递归思想与栈类似，当前结点只有在它的左右子树全部完成操作才会返回到上一层父亲结点，也意味着在此之前它这一层的相关操作(当前结点的高度)会保存。
  N叉树更简单一些，依旧是DFS。根据题设给的树的结构体，每个结点都会有自己的孩子结点集合vector<Node*> children。因此，我们只需遍历求出当前结点的孩子结点集合的深度最大值，最后+1返回即可。

二叉树

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(!root)
            return 0;
        int left_depth = maxDepth(root->left);
        int right_depth = maxDepth(root->right);

        return max(left_depth, right_depth) + 1;
    }
};

N叉树

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> children;

    Node() {}

    Node(int _val) {
        val = _val;
    }

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
public:
    int maxDepth(Node* root) {
        if(!root)
            return 0;
        int result = 0;
        for(auto iter : root->children){
            result = max(result, maxDepth(iter));
        }
        return result + 1;
    }
};

如果有错误或者不严谨的地方，请务必给予指正，十分感谢。
本人blog：http://breadhunter.gitee.io