diff options
author | Santo Cariotti <santo@dcariotti.me> | 2024-05-28 10:29:13 +0200 |
---|---|---|
committer | Santo Cariotti <santo@dcariotti.me> | 2024-05-28 10:29:13 +0200 |
commit | f05d888a0b621ca4e99e2b0fb6e23c097006fe41 (patch) | |
tree | eebbb2489144112d3288393e354d19375a0aa088 /progs/dont_care/a730.py |
Init
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 |