完全二叉树的节点个数-力扣

完全二叉树 的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。
因此，完全二叉树只有叶子节点一层的右侧缺少节点，对树的左节点和右节点进行递归，如果两者深度相同，则这棵树的节点数为2^k - 1，否则对左右子树进行判断，这样即可不用遍历整个树中符合完全二叉树的子树部分就。

• 注，这里的指数运算使用了位运算符，而不是 ^，一开始使用 ^ 运算符，返回的结果与预想的一直不对，这个运算符是对两个数按位进行异或，而不是幂运算。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:

    int countNodes(TreeNode* root) {
        if(root == nullptr){
            return 0;
        }
        TreeNode * left = root->left;
        TreeNode * right = root->right;
        int l_depth = 0;
        int r_depth = 0;
        while(left != nullptr){
            l_depth++;
            left = left->left;           
        }
        while(right != nullptr){
            r_depth++;
            right = right->right;
        }
        if(l_depth == r_depth){
            return (2 << l_depth) - 1;
        }

        return countNodes(root->left) + countNodes(root->right) + 1;
    }
};