【leetcode】222 完全二叉树的节点个数（二叉树，二分查找）

题目链接：https://leetcode-cn.com/problems/count-complete-tree-nodes/

题目描述

给出一个完全二叉树，求出该树的节点个数。

说明：

完全二叉树的定义如下：在完全二叉树中，除了最底层节点可能没填满外，其余每层节点数都达到最大值，并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层，则该层包含 1~ 2h 个节点。

示例:

输入: 
    1
   / \
  2   3
 / \  /
4  5 6

输出: 6

思路

1 层序遍历

/*
 * 层序遍历记录节点数
 */
class Solution {
public:
    int countNodes(TreeNode* root) {
        if(!root) return 0;
        queue<TreeNode*> q;
        int ret = 0;
        q.push(root);
        while (!q.empty()){
            TreeNode *node = q.front();
            ++ret;
            q.pop();
            if(node->left)
                q.push(node->left);
            if(node->right)
                q.push(node->right);
        }
        return ret;
    }
};

2 递归

/*
 * 递归
 */
class Solution {
public:
    int countNodes(TreeNode* root) {
        if(!root) return 0;
        int leftNodes = countNodes(root->left);
        int rightNodes = countNodes(root->right);
        return leftNodes + rightNodes + 1;
    }
};

3 递归+完全二叉树性质 O（log n）

前面两种方法都是暴力方法，时间复杂度为O（n），没有利用到完全二叉树的性质。
注意：完全二叉树的子树也肯定是完全二叉树
完全二叉树的节点数可以直接计算，所以我们采用二分查找的方差，递归地去计算不是完全二叉树的子树节点数。

二分搜索+递归：
（1）先判断左子树是不是完全二叉树，如果是的话通过树的高度可以计算得到左子树的节点数，总节点数=左子树节点数+1+右子树节点数（递归计算）
（2）如果不是，则右子树必然是完全二叉树，通过高度可以得到右子树的节点数，总节点数=右子树节点数+1+左子树节点数（递归计算）

复杂度分析

• 时间复杂度：O（log n）

/*
 * 递归+完全二叉树性质
 * O(log N)
 */
class Solution {
public:
    int countNodes(TreeNode* root) {
        if(!root) return 0;
        int h = height(root); 
        if(h == height(root->right) + 1) // 右子树高度=根高度-1，左子树一定是完全二叉树
            return (1<< (h-1)) + countNodes(root->right);
        else // 右子树高度=根高度-2
            return (1 << (h-2)) + countNodes(root->left);;
    }

private:
    // 完全二叉树高度
    int height(TreeNode *root){
        int count = 0;
        while (root){
            root = root->left;
            count ++;
        }
        return count;
    }
};