Advertisement
nikunjsoni

222-2

May 5th, 2021
99
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.80 KB | None | 0 0
  1. /**
  2.  * Definition for a binary tree node.
  3.  * struct TreeNode {
  4.  *     int val;
  5.  *     TreeNode *left;
  6.  *     TreeNode *right;
  7.  *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
  8.  *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
  9.  *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
  10.  * };
  11.  */
  12. class Solution {
  13. public:
  14.     int countNodes(TreeNode* root) {
  15.         if(root==NULL) return 0;
  16.         TreeNode *lnode=root,*rnode=root;
  17.         int l=0,r=0;
  18.         while(lnode){
  19.             l++;
  20.             lnode=lnode->left;
  21.         }
  22.          while(rnode){
  23.             r++;
  24.             rnode=rnode->right;
  25.         }
  26.         if(l==r) return pow(2,l)-1;
  27.         return countNodes(root->left)+countNodes(root->right)+1;
  28.        
  29.     }
  30. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement