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

文章初心

笔者是应届生，二叉树的遍历以及二叉树的属性等对于面试来说，显得尤为重要，今天用一道题来巩固二叉树的遍历和二叉树的属性的知识。

难易程度

• middle

题目考察

• 二叉树的遍历
• 递归调用
• 栈的使用
• 满二叉树的性质

深度优先遍历（后序遍历）

思路：先递归左子树的节点个数，再递归右子树的节点个数，最后返回左右子树节点个数 + 1（根节点自身）

//时间复杂度O(nlogn)   空间复杂度O(n)
class Solution {
public:
    int getNum(TreeNode* node) {
        int num = 0;
        if (node == nullptr) return num;
        int getLeftNum = getNum(node->left); //左
        int getRightNum = getNum(node->right); //右
         num = getLeftNum + getRightNum + 1; //中
        return num;
    }
    int countNodes(TreeNode* root){
        return getNum(root);
    }
};

（优化1）层序遍历

思路：使用层序遍历，通过辅助队列的使用，每遍历一次，num++，最终返回节点个数（num值）

 //时间复杂度O(n)   空间复杂度O(n)
class Solution {
public:
    int countNodes(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != nullptr) que.push(root);
        int num = 0;
        while (!que.empty()) {
            int size = que.size();
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                num++;
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
        }
        return num;
    }
};

（优化2）深度优先遍历（后序遍历）

思路：利用满二叉树的性质，满二叉树的节点个数为2 * n -1，n代表层数，这样遍历就不需要借助额外的辅助数据结构，递归也是遍历二叉树本身，时间复杂度 (logn) ^ 2 和空间复杂度logn也能得到优化

//时间复杂度O(logn) ^ 2   空间复杂度O(logn)
//满二叉树的节点个数为2 * n -1，n代表层数 
class Solution {
public:
    int countNodes(TreeNode* root) {
        if (root == nullptr) return 0;
        TreeNode* left = root->left;
        TreeNode* right = root->right;
        int leftHight = 0, rightHight = 0;
        while (left) {  //递归每一个左节点
            left = left->left;
            leftHight++;
        }
        while (right) {  //递归每一个右节点
            right = right->right;
            rightHight++;
        }
        if (leftHight == rightHight) { //利用满二叉树的性质
            return (2 << leftHight) - 1; //注意(2<<1) 相当于2^2，所以leftHeight初始为0
        }
        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};

总结

通过本题可以发现，使用深度优先遍历的时间复杂度稍微有点大，然后使用了层序遍历的方法进行优化，但是空间复杂度还是可以优化的，进而利用了满二叉树的性质把这道题优化到极致，以后类似的二叉树的题，这样的思路一定有所帮助，加油！