ধরুন আমাদের একটি সাজানো অ্যারে রয়েছে। আমাদের একটি উচ্চতা-ভারসাম্যযুক্ত বাইনারি অনুসন্ধান তৈরি করতে হবে। এই সমস্যায়, একটি উচ্চতা-ভারসাম্যযুক্ত বাইনারি ট্রি আসলে একটি বাইনারি ট্রি যেখানে প্রতিটি নোডের দুটি সাবট্রির গভীরতা কখনোই 1 এর বেশি হয় না। ধরুন অ্যারেটি [-10, -3, 0, 5, 9 এর মতো। ]। সুতরাং একটি সম্ভাব্য আউটপুট এরকম হবে:[0, -3, 9, -10, নাল, 5]
এটি সমাধান করতে, আমরা এই পদক্ষেপগুলি অনুসরণ করব৷
- যদি A খালি হয়, তাহলে Null ফেরত দিন
- মাঝের উপাদানটি খুঁজুন এবং এটিকে রুট করুন
- অ্যারেটিকে দুটি সাব-অ্যারেতে ভাগ করুন, মধ্য এলিমেন্টের বাম অংশ এবং মধ্য এলিমেন্টের ডান অংশ
- বাম সাবারে এবং ডান সাবয়ারের জন্য পুনরাবৃত্তিমূলকভাবে একই কাজ সম্পাদন করুন।
আরো ভালোভাবে বোঝার জন্য নিচের বাস্তবায়নটি দেখি -
উদাহরণ
class TreeNode: def __init__(self, data, left = None, right = None): self.data = data self.left = left self.right = right def insert(temp,data): que = [] que.append(temp) while (len(que)): temp = que[0] que.pop(0) if (not temp.left): if data is not None: temp.left = TreeNode(data) else: temp.left = TreeNode(0) break else: que.append(temp.left) if (not temp.right): if data is not None: temp.right = TreeNode(data) else: temp.right = TreeNode(0) break else: que.append(temp.right) def make_tree(elements): Tree = TreeNode(elements[0]) for element in elements[1:]: insert(Tree, element) return Tree def height(root): if root is None: return 0 else : # Compute the height of left and right subtree l_height = height(root.left) r_height = height(root.right) #Find the greater one, and return it if l_height > r_height : return l_height+1 else: return r_height+1 def print_given_level(root, level): if root is None: return if level == 1: print(root.data,end = ',') elif level > 1 : print_given_level(root.left , level-1) print_given_level(root.right , level-1) def level_order(root): print('[', end = '') h = height(root) for i in range(1, h+1): print_given_level(root, i) print(']') class Solution(object): def sortedArrayToBST(self, nums): """ :type nums: List[int] :rtype: TreeNode """ if not nums: return None mid = nums[len(nums)//2] root = TreeNode(mid) root.left = self.sortedArrayToBST(nums[:len(nums)//2]) root.right = self.sortedArrayToBST(nums[len(nums)//2 +1 :]) return root nums = [-10,-3,0,5,9] ob1 = Solution() bst = ob1.sortedArrayToBST(nums) level_order(bst)
ইনপুট
nums = [-10,-3,0,5,9]
আউটপুট
[0,-3,9,-10,5,]