diff options
Diffstat (limited to 'progs/dont_care/a730.py')
| -rw-r--r-- | progs/dont_care/a730.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/progs/dont_care/a730.py b/progs/dont_care/a730.py new file mode 100644 index 0000000..ba741d2 --- /dev/null +++ b/progs/dont_care/a730.py @@ -0,0 +1,18 @@ +class Node:
+ def __init__(self, data):
+ self.data = data
+ self.left = None
+ self.right = None
+def get_height(root):
+ if root is None:
+ return 0
+ return max(get_height(root.left), get_height(root.right)) + 1
+def is_tree_balanced(root):
+ if root is None:
+ return True
+ lh = get_height(root.left)
+ rh = get_height(root.right)
+ if (abs(lh - rh) <= 1) and is_tree_balanced(
+ root.left) is True and is_tree_balanced( root.right) is True:
+ return True
+ return False
\ No newline at end of file |