ধরুন আমাদের একটি বাইনারি সার্চ ট্রি আছে। আমাদের একই নোড ভ্যালু সহ একটি ব্যালেন্সড বাইনারি সার্চ ট্রি খুঁজে বের করতে হবে। একটি BST বলা হয় ব্যালেন্সড যদি এবং কেবল যদি প্রতিটি নোডের দুটি সাবট্রির গভীরতা ১-র বেশি পার্থক্য না রাখে। একাধিক ফলাফল থাকলে যেকোনো একটি রিটার্ন করুন। ট্রি যদি নিচের মতো হয় −
সমাধানের ধাপগুলো:
inorder() মেথড ডিফাইন করুন, এটি ইনঅর্ডার ট্রাভার্সাল সিকোয়েন্স একটি অ্যারেতে স্টোর করবে
construct() মেথড ডিফাইন করুন, যা low এবং high নেবে −
যদি low > high হয়, তবে null রিটার্ন করুন
mid := low + (high - low) / 2
root := arr[mid] ভ্যালু দিয়ে নতুন নোড
root-এর লেফট := construct(low, mid – 1) এবং রাইট := construct(mid + 1, high)
root রিটার্ন করুন
মেইন মেথড থেকে inorder কল করুন এবং construct(0, arr.size() - 1) রিটার্ন করুন
উদাহরণ (C++)
নিচের ইমপ্লিমেন্টেশন দেখুন ভালো বুঝতে −
#include <bits/stdc++.h> using namespace std; class TreeNode{ public: int val; TreeNode *left, *right; TreeNode(int data){ val = data; left = right = NULL; } }; void insert(TreeNode **root, int val){ queue<TreeNode*> q; q.push(*root); while(q.size()){ TreeNode *temp = q.front(); q.pop(); if(!temp->left){ if(val != NULL) temp->left = new TreeNode(val); else temp->left = new TreeNode(0); return; }else{ q.push(temp->left); } if(!temp->right){ if(val != NULL) temp->right = new TreeNode(val); else temp->right = new TreeNode(0); return; }else{ q.push(temp->right); } } } TreeNode *make_tree(vector<int> v){ TreeNode *root = new TreeNode(v[0]); for(int i = 1; i<v.size(); i++){ insert(&root, v[i]); } return root; } void tree_level_trav(TreeNode*root){ if (root == NULL) return; cout << "["; queue<TreeNode *> q; TreeNode *curr; q.push(root); q.push(NULL); while (q.size() > 1) { curr = q.front(); q.pop(); if (curr == NULL){ q.push(NULL); } else { if(curr->left) q.push(curr->left); if(curr->right) q.push(curr->right); if(curr->val == 0 || curr == NULL){ cout << "null" << ", "; }else{ cout << curr->val << ", "; } } } cout << "]"<<endl; } class Solution { public: vector <int> arr; void inorder(TreeNode* node){ if(!node || node->val == 0) return; inorder(node->left); arr.push_back(node->val); inorder(node->right); } TreeNode* construct(int low, int high){ if(low > high) return NULL; int mid = low + (high - low) / 2; TreeNode* root = new TreeNode(arr[mid]); root->left = construct(low, mid - 1); root->right = construct(mid + 1, high); return root; } TreeNode* balanceBST(TreeNode* root) { inorder(root); return construct(0, (int)arr.size() - 1); } }; main(){ vector<int> v = {1,NULL,2,NULL,NULL,NULL,3,NULL,NULL,NULL,NULL,NULL,NULL,NULL,4}; TreeNode *root = make_tree(v); Solution ob; tree_level_trav(ob.balanceBST(root)); }
ইনপুট
[1,NULL,2,NULL,NULL,NULL,3,NULL,NULL,NULL,NULL,NULL,NULL,NULL,4]
আউটপুট
[2, 1, 3, 4, ]
