ধরুন আমাদের একটি বাইনারি ট্রি আছে, যেখানে প্রতিটি নোডের মান হয় একটি 0 বা একটি 1। আমাদের একই ট্রি খুঁজে বের করতে হবে যেখানে 1 নেই এমন প্রতিটি সাবট্রি মুছে ফেলা হয়েছে। তাই গাছটি যদি −
এর মত হয়
এটি সমাধান করতে, আমরা এই পদক্ষেপগুলি অনুসরণ করব -
-
একটি পুনরাবৃত্ত পদ্ধতি সংজ্ঞায়িত করুন solve(), এটি নোড নেবে। পদ্ধতিটি −
এর মত হবে -
যদি নোডটি নাল থাকে, তাহলে নাল ফেরত দিন
-
নোডের বাম :=সমাধান (নোডের বাম)
-
নোডের অধিকার :=সমাধান (নোডের ডান)
-
যদি নোডের বাম অংশ নাল হয় এবং নোডের ডানদিকেও নাল হয় এবং নোডের মান 0 হয়, তাহলে নাল ফেরত দিন
-
রিটার্ন নোড
আরো ভালোভাবে বোঝার জন্য আসুন নিচের বাস্তবায়ন দেখি -
উদাহরণ
#include <bits/stdc++.h>
using namespace std;
class TreeNode{
public:
int val;
TreeNode *left, *right;
TreeNode(int data){
val = data;
left = NULL;
right = NULL;
}
};
void inorder(TreeNode *root){
if(root){
inorder(root->left);
cout << root->val << ", ";
inorder(root->right);
}
}
class Solution {
public:
TreeNode* pruneTree(TreeNode* node) {
if(!node)return NULL;
node->left = pruneTree(node->left);
node->right = pruneTree(node->right);
if(!node->left && !node->right && !node->val){
return NULL;
}
return node;
}
};
main(){
TreeNode *root = new TreeNode(1);
root->left = new TreeNode(1);
root->right = new TreeNode(0);
root->left->left = new TreeNode(1);
root->left->right = new TreeNode(1);
root->right->left = new TreeNode(0);
root->right->right = new TreeNode(1);
root->left->left->left = new TreeNode(0);
Solution ob;
inorder(ob.pruneTree(root));
} ইনপুট
TreeNode *root = new TreeNode(1); root−>left = new TreeNode(1); root−>right = new TreeNode(0); root−>left−>left = new TreeNode(1); root−>left−>right = new TreeNode(1); root−>right−>left = new TreeNode(0); root−>right−>right = new TreeNode(1); root−>left−>left−>left = new TreeNode(0);
আউটপুট
1, 1, 1, 1, 0, 1,
