ধরুন আমাদের একটি বাইনারি গাছের মূল আছে, এর মূলের স্তর হল 1, এর বাচ্চাদের স্তর হল 2, ইত্যাদি। আমাদের ক্ষুদ্রতম স্তর X ফেরত দিতে হবে যাতে X স্তরে নোডের সমস্ত মানের যোগফল সর্বাধিক হয়। তাই গাছটি যদি −
এর মত হয়
তারপর আউটপুট হবে 2, স্তর 1 যোগফল =1, স্তর 2 যোগফল 7 + 0 =7, স্তর 2 যোগফল হল 7 + (-8) =-1, তাই সর্বোচ্চ হল স্তরের 2, তাই আউটপুট হল 2।
এটি সমাধান করতে, আমরা এই পদক্ষেপগুলি অনুসরণ করব -
- স্তর :=1, যোগফল :=r এর মান, ansLevel :=স্তর, ansSum :=যোগফল
- একটি সারি q সংজ্ঞায়িত করুন, প্রদত্ত নোড r q এ সন্নিবেশ করুন
- যখন q খালি নয়
- ক্ষমতা :=q এর আকার
- লেভেল 1 দ্বারা বাড়ান, যোগফল :=0
- যদিও ক্ষমতা 0
- নয়
- নোড :=q থেকে সামনের নোড, q থেকে নোড মুছে দিন
- যদি নোডের ডানটি বৈধ হয়, তাহলে sum :=sum + ডান নোডের মান, q তে ডান নোড সন্নিবেশ করুন
- যদি নোডের বাম অংশটি বৈধ হয়, তাহলে sum :=sum + বাম নোডের মান, q এ বাম নোড সন্নিবেশ করুন
- ক্ষমতা 1 দ্বারা হ্রাস করুন
- যদি ansSum
- আন্সলেভেল ফেরত দিন
উদাহরণ(C++)
আরো ভালোভাবে বোঝার জন্য নিচের বাস্তবায়নটি দেখি -
#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 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; } class Solution { public: int maxLevelSum(TreeNode* r) { int level = 1, sum = r->val; int ansLevel = level, ansSum = sum; queue <TreeNode*> q; q.push(r); while(!q.empty()){ int capacity = q.size(); level++; sum = 0; while(capacity--){ TreeNode* node = q.front(); q.pop(); if(node->right){ sum += node->right->val; q.push(node->right); } if(node->left){ sum += node->left->val; q.push(node->left); } } if(ansSum<sum){ ansSum = sum; ansLevel = level; } } return ansLevel; } }; main(){ vector<int> v = {1,7,0,7,-8,NULL,NULL}; TreeNode *root = make_tree(v); Solution ob; cout <<ob.maxLevelSum(root); }
ইনপুট
[1,7,0,7,-8,null,null]
আউটপুট
2