Leetcode 559. N叉树的最大深度(DFS和BFS解法)

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

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

例如,给定一个 3叉树 :

返回其最大深度,3。

说明:

  1. 树的深度不会超过 1000
  2. 树的节点总不会超过 5000

【DFS解法】 

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

    Node() {}

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/
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;
    }
};

【BFS解法】

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

    Node() {}

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

猜你喜欢

转载自blog.csdn.net/zpznba/article/details/88073062