LeetCode 222. Count Complete Tree Nodes
complete binary tree:除最后一行外每一行的节点都有两个儿子,最后一行的节点尽可能靠左。
ver0:
1 class Solution { 2 public: 3 int countNodes(TreeNode* root) { 4 if(!root) return 0; 5 return 1 + countNodes(root->left) + countNodes(root->right); 6 } 7 };
不出意料地TLE。
ver1:
1 class Solution { 2 public: 3 int countNodes(TreeNode* root) { 4 if(!root) return 0; 5 TreeNode* l = root, * r = root; 6 int llen = 0, rlen = 0; 7 // while(l->left){//ERROR! CORRECT:while(l) 8 // ++llen; 9 // l = l->left; 10 // } 11 // while(r->right){ 12 // ++rlen; 13 // r = r->right; 14 // } 15 for(; l!=NULL; ++llen, l=l->left); //ATTENTION! 16 for(; r!=NULL; ++rlen, r=r->right); 17 return llen == rlen ? (1<<llen) - 1 : 1 + countNodes(root->left) + countNodes(root->right);//ATTENTION 18 19 20 } 21 };
根据最后一行的节点尽可能靠左这一特点,先计算root->left->left->...这一路径长和root->right->right->...这一路径长。如果相等,则整棵树都是满的,利用等比数列公式(注意用了位运算);不相等则递归。