完全二叉树结点总数/leetcode 222. 完全二叉树的节点个数（二分）

题目描述
给定一棵完全二叉树的头节点head，返回这棵树的节点个数。如果完全二叉树的节点数为N，请实现时间复杂度低于O(N)的解法。
示例1
输入 {1,2,3}
返回值 3

满二叉树节点的计算方法：如果一个二叉树的层数为K，且结点总数是(2^k) -1。

• 如果左右子树层数相同，那么可以按照满二叉树公式计算节点；
• 如果左右子树层数不同，那么需要一直向下遍历，该棵树的节点个数等于1+左子树节点数+右子树节点数。此时与普通二叉树的计算方式类似，只是如果是左右子树高度相等的话，那么可以直接计算，不需要递归了。

可以看做是一种特殊的二分，需要二分 log_n 次，每次要从左右遍历 log_n 次，时间复杂度为 log²_n

二分版本：

class Solution {
public: 
    int nodeNum(struct TreeNode* head) {
       if(!head) return 0;
       auto p =head,q=head;
       int leftH=0,rightH=0;
       while(p) 
       {
           p=p->left;
           leftH++;
       }
        while(q) 
       {
           q=q->right;
           rightH++;
       }
       if(leftH==rightH) return pow(2,leftH)-1;
       return nodeNum(head->left)+nodeNum(head->right)+1;
    }
};

以下两个版本同样适用于普通的二叉树求节点数量。
递归版本：

class Solution {
public:
    int nodeNum(struct TreeNode* head) {
        if(!head) return 0;
        int left = nodeNum(head->left);
        int right= nodeNum(head->right);
        return left+right+1;
    }
};

迭代版本：

class Solution {
public:
    int nodeNum(struct TreeNode* head) {
        queue<TreeNode*> q;
        int res=0;
        if(head)  q.push(head);
        while(q.size())
        {
            TreeNode* t=q.front();q.pop();
            res++;
            if(t->left)  q.push(t->left);           
            if(t->right)  q.push(t->right);

        }
        return res;
    }
};