[leetcode] Balanced Binary Tree

 1 /** 
 2  * Definition for binary tree 
 3  * struct TreeNode { 
 4  *     int val; 
 5  *     TreeNode *left; 
 6  *     TreeNode *right; 
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} 
 8  * }; 
 9  */ 
10 class Solution {  
11     public:  
12         bool isBalanced(TreeNode *root) {  
13             // Note: The Solution object is instantiated only once and is reused by each test case.  
14             int depth = 0;  
15             return isbalance(root, depth);  
16         }  
17         bool isbalance(TreeNode *root, int &depth)  
18         {  
19             if(root == NULL)  
20             {  
21                 depth = 0;  
22                 return true;  
23             }  
24             int ld,rd;  
25             if( isbalance(root->left,ld) && isbalance(root->right,rd))  
26             {  
27                 if( abs(ld - rd) > 1)  
28                 {  
29                     return false;  
30                 }  
31                 depth = ld > rd ? ld + 1 : rd + 1;  
32                 return true;  
33             }
34             else return false;
35         }  
36 };