leetcode - Balanced Binary Tree

 1 #include <stddef.h>
 2 #include <iostream>
 3 /**
 4  * Definition for binary tree
 5  * struct TreeNode {
 6  *     int val;
 7  *     TreeNode *left;
 8  *     TreeNode *right;
 9  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
10  * };
11  */
12 
13 struct TreeNode
14 {
15     int val;
16     TreeNode *left;
17     TreeNode *right;
18     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
19 };
20 
21 class Solution
22 {
23 public:
24     bool isBalanced(TreeNode *root)
25     {
26         balanced = true;
27         getDepth(root);
28 
29         return balanced;
30     }
31     int getDepth(TreeNode *root)
32     {
33         if (root == NULL)
34         {
35             return 0;
36         }
37 
38         int leftDepth = getDepth(root->left);
39         int rightDepth = getDepth(root->right);
40 
41         if (leftDepth - rightDepth > 1 || leftDepth - rightDepth < -1)
42         {
43             balanced = false;
44         }
45 
46         return leftDepth > rightDepth ? leftDepth + 1 : rightDepth + 1;
47     }
48 private:
49     bool balanced;
50 };
51 
52 int main()
53 {
54     TreeNode *root = new TreeNode(1);
55     root->right = new TreeNode(1);
56     root->right->right = new TreeNode(1);
57     Solution s;
58     s.isBalanced(root);
59     std::cout << root->val << std::endl << root->right->val << std::endl << root->right->val << std::endl;
60     system("pause");
61 
62     return 0;
63 }