代码随想录算法训练营Day16 || leetCode 104.二叉树的最大深度 || 111.二叉树的最小深度 || 559.n叉树的最大深度 || 222.完全二叉树的节点个数

104.二叉树的最大深度  111.二叉树的最小深度

这两题的答案在昨天的文章中已经写过，在此不做赘述。

559.n叉树的最大深度 

和二叉树深度代码一样，注意修改TreeNode为Node，以及children[i]中i的范围。

class Solution {
public:
    int maxDepth(Node* root) {
        if (root == NULL) return 0;
        int depth = 0;
        queue<Node*> que;
        que.push(root);
        while(!que.empty()) {
            int size = que.size();
            depth++; // 记录深度
            for (int i = 0; i < size; i++) {
                Node* node = que.front();
                que.pop();
                for(int i = 0; i < node->children.size(); i++){
                    if (node->children[i]) que.push(node->children[i]);
                }
            }
        }
        return depth;
    }
};

222.完全二叉树的节点个数

最简单的递归法，后续可根据二叉树性质来优化

class Solution {
private:
    int getNodesNum(TreeNode* cur) {
        if (cur == NULL) return 0;
        int leftNum = getNodesNum(cur->left);      // 左
        int rightNum = getNodesNum(cur->right);    // 右
        int treeNum = leftNum + rightNum + 1;      // 中
        return treeNum;
    }
public:
    int countNodes(TreeNode* root) {
        return getNodesNum(root);
    }
};