Leetcode 559. Maximum Depth of N-ary Tree

Leetcode 559. Maximum Depth of N-ary Tree

c++,如果本节点为空,返回0,否则返回 这棵树孩子中(找到每个节点的最大值,返回最大值+1即可,1是本节点的深度)

/*
// 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 ans = 0;
        for(int i = 0; i < root->children.size(); i++)
            ans = max(ans, maxDepth(root->children[i]));
        return ans + 1;
    }
};