Update 222.Count-Complete-Tree-Nodes_v2.cpp

KristinHuangg · Nov 9, 2019 · 282afbe · 282afbe
1 parent 0cec913
commit 282afbe
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Tree/222.Count-Complete-Tree-Nodes/222.Count-Complete-Tree-Nodes_v2.cpp b/Tree/222.Count-Complete-Tree-Nodes/222.Count-Complete-Tree-Nodes_v2.cpp
@@ -11,15 +11,14 @@ class Solution {
 public:
     int countNodes(TreeNode* root) 
     {
-        if (root==NULL) return 0;
         int h = 0;
         TreeNode* node = root;
         while (node!=NULL)
         {
             h++;
             node = node->left;            
         }
-        int low = 1;
+        int low = (1<<(h-1));
         int hi = (1<<h)-1;
 
         while (low<hi)