C++ (Cpp) getRightDepthの例

コード例 #1

ファイル: 222_Count_Complete_Tree_Nodes.cpp プロジェクト: heshenghuan/leetcode-exercise

 int countNodes(TreeNode* root) {
     if(!root) return 0;
     
     int l = getLeftDepth(root);
     int r = getRightDepth(root);
     
     if(l == r) return (2<<(l-1)) - 1;
     
     return countNodes(root->left) + countNodes(root->right) + 1;
 }

コード例 #2

ファイル: 222.cpp プロジェクト: HaochenLiu/My-LeetCode-CPP

 int countNodes(TreeNode* root) {
     if(root == NULL) return 0;
     if(root->left == NULL && root->right == NULL) return 1;
     int l = getLeftDepth(root);
     int r = getRightDepth(root);
     if(l == r) {
         return (1<<l) - 1;
     } else {
         return 1 + countNodes(root->left) + countNodes(root->right);
     }
 }