-
Notifications
You must be signed in to change notification settings - Fork 0
/
isBalanced.cc
60 lines (58 loc) · 1.65 KB
/
isBalanced.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool isBalanced(TreeNode *root) {
if (!root) return true;
int h;
return _isBal(root, h);
}
bool _isBal(TreeNode *root, int &h) {
if (!root) {
h = 0;
return true;
}
int h1, h2;
bool l = _isBal(root->left, h1);
bool r = _isBal(root->right, h2);
h = max(h1, h2) + 1;
if (l && r && (abs(h1 - h2) <= 1)) return true;
else return false;
}
//--------------------------------------//
bool ret = true;
bool isBalanced(TreeNode *root) {
if (!root) return true;
int left = depth(root->left);
int right = depth(root->right);
if (abs(left - right) > 1) return false;
else return ret;
}
int depth(TreeNode *root) {
if (!root) return 0;
int left = depth(root->left);
int right = depth(root->right);
if (abs(left - right) > 1) ret = false;
return max(left, right) + 1;
}
/*bool isBalanced(TreeNode *root) {
if (!root) return true;
int left = depth(root->left);
int right = depth(root->right);
if (abs(left - right) > 1) return false;
return isBalanced(root->left) && isBalanced(root->right);
}
int depth(TreeNode *root) {
if (!root) return 0;
int left = depth(root->left);
int right = depth(root->right);
return left > right ? left + 1 : right + 1;
}*/
};