LeetCode222 Count Complete Tree Nodes

题目链接：

https://leetcode.com/problems/count-complete-tree-nodes/

题目描述：

求一个完全二叉树的节点个数。

题目分析：

需要注意的是遍历的方式求解节点个数会超时，时间复杂度为O(n)。利用完全二叉树与满二叉树节点个数的特性。满二叉树的节点个数为2^h-1，h为树高。先判断该完全二叉树是不是满二叉树，如果是直接返回结果，如果不是再递归求解。

如果节点左子树最左边的深度，与节点右子树最右边的深度相等，则以该节点为根节点的树是满二叉树，如果不等，则以该节点为根节点的树的节点个数=其左子树的节点个数+其右子树的节点个数+本身。

代码：

class Solution {
public:
    int countNodes(TreeNode* root) {
        if(root==NULL){
            return 0;
        }
        int leftHgt=1;
        int rightHgt=1;
        TreeNode* ptr=NULL;
        ptr=root->left;
        while(ptr!=NULL){
            leftHgt++;
            ptr=ptr->left;
        }
        ptr=root->right;
        while(ptr!=NULL){
            rightHgt++;
            ptr=ptr->right;
        }
        if(leftHgt==rightHgt){
            return (1<<leftHgt)-1;
        }
        else{
            return countNodes(root->left)+countNodes(root->right)+1;
        }
    }
};