From f05d888a0b621ca4e99e2b0fb6e23c097006fe41 Mon Sep 17 00:00:00 2001 From: Santo Cariotti Date: Tue, 28 May 2024 10:29:13 +0200 Subject: Init --- progs/dont_care/a730.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 progs/dont_care/a730.py (limited to 'progs/dont_care/a730.py') 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 -- cgit v1.2.3-18-g5258