ค้นหาค่าที่มากที่สุดในแต่ละแถวต้นไม้ใน C++

สมมติว่าเรามีต้นไม้ไบนารี เราต้องหาองค์ประกอบที่ใหญ่ที่สุดของแต่ละระดับของต้นไม้นั้น ดังนั้นถ้าต้นไม้เป็นเหมือน −

จากนั้นผลลัพธ์จะเป็น [3,5,8]

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -

• กำหนดอาร์เรย์ที่เรียกว่า ans

• กำหนดฟังก์ชันแบบเรียกซ้ำ Solvent() จะใช้โหนดทรี และระดับ ระดับเริ่มต้นเป็น 0 วิธีการนี้จะทำหน้าที่เหมือน -

• หากโหนดเป็นโมฆะ ให้ส่งคืน

• ถ้าระดับ =ขนาดของ ans ให้แทรกค่าโหนดลงใน ans มิฉะนั้น ans[level] :=max of ans[level] และค่าโหนด

• แก้การโทร (ทรีย่อยด้านซ้ายของโหนด ระดับ + 1)

• แก้การโทร (แผนผังย่อยด้านขวาของโหนด ระดับ + 1)

• จากเมธอดหลัก เรียก Solve() โดยใช้รูทเป็นพารามิเตอร์ และระดับ =0

• แล้วตอบกลับ

ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -

ตัวอย่าง

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<auto> v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << v[i] << ", ";
   }
   cout << "]"<<endl;
}
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:
   vector <int> ans;
   void solve(TreeNode* node, int level = 0){
      if(!node)return;
      if(level == ans.size()){
         ans.push_back(node->val);
      } else {
         ans[level] = max(ans[level], node->val);
      }
      solve(node->left, level + 1);
      solve(node->right, level + 1);
   }
   vector<int> largestValues(TreeNode* root) {
      solve(root);
      return ans;
   }
};
main(){
   vector<int> v = {1,3,2,5,3,NULL,9};
   TreeNode *tree = make_tree(v);
   Solution ob;
   print_vector(ob.largestValues(tree));
}

อินพุต

[1,3,2,5,3,null,9]

ผลลัพธ์

[1, 3, 9, ]