forked from shichao-an/leetcode-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsolution.py
More file actions
24 lines (22 loc) · 694 Bytes
/
solution.py
File metadata and controls
24 lines (22 loc) · 694 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# Definition for a binary tree node
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# @param root, a tree node
# @return a boolean
def isBalanced(self, root):
if root is None:
return True
else:
if self.isBalanced(root.left) and self.isBalanced(root.right):
return abs(self.depth(root.left) - self.depth(root.right)) <= 1
else:
return False
def depth(self, root):
if root is None:
return -1
else:
return max(self.depth(root.left), self.depth(root.right)) + 1