LeetCode 222. Count Complete Tree Nodes

complete binary tree:除最后一行外每一行的节点都有两个儿子，最后一行的节点尽可能靠左。

ver0：

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

不出意料地TLE。

ver1：

class Solution {
public:
int countNodes(TreeNode* root) {
if(!root) return 0;
TreeNode* l = root, * r = root;
int llen = 0, rlen = 0;
// while(l->left){//ERROR! CORRECT:while(l)
//     ++llen;
//     l = l->left;
// }
// while(r->right){
//     ++rlen;
//     r = r->right;
// }
for(; l!=NULL; ++llen, l=l->left); //ATTENTION!
for(; r!=NULL; ++rlen, r=r->right);
return llen == rlen ? (1<<llen) - 1 : 1 + countNodes(root->left) + countNodes(root->right);//ATTENTION

}
};

根据最后一行的节点尽可能靠左这一特点，先计算root->left->left->...这一路径长和root->right->right->...这一路径长。如果相等，则整棵树都是满的，利用等比数列公式（注意用了位运算）；不相等则递归。