Leetcode 559. N叉樹的最大深度

韓豆豆小姐發表於2020-10-15

給定一個 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 dfs(Node *root){
        if(!root) return 0;
        int res=0;//這裡不能設定為全域性變數
        for(int i=0;i<root->children.size();i++){
             res =max(dfs(root->children[i]),res);
        }
        return res+1;
    }
    int maxDepth(Node* root) {
       int ans = dfs(root);
       return ans;
    }
};

思路二:利用層次遍歷求深度

/*
// 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==NULL) return 0;
        queue<Node*> que;
        if (root != NULL) que.push(root);
        int depth = 0; 
        while (!que.empty()) {
            int size = que.size();//這裡簡直了
            depth++; // 記錄深度
            //寫i<que.size()就無法通過,也是奇怪了
            for (int i = 0; i < size; i++) {
                Node* node = que.front();
                que.pop();
                for (auto it:node->children) {
                    que.push(it);
                }
            }
        }
        return depth;
    }
};

相關文章