ค่าเฉลี่ยของระดับในไบนารีทรีใน C++

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

ดังนั้นหากอินพุตเป็นแบบ

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

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

• กำหนดผลลัพธ์อาร์เรย์

• กำหนดหนึ่งคิว q

• แทรกรูทลงใน q

• ในขณะที่ (ไม่ใช่ q ว่างเปล่า) ทำ -

  • n :=ขนาดของ q

  • กำหนดอุณหภูมิอาร์เรย์

  • ในขณะที่ n ไม่ใช่ศูนย์ ให้ทำ -

    • t :=องค์ประกอบแรกของ q

    • ใส่ค่าของ t ลงใน temp

    • ลบองค์ประกอบออกจาก q

    • ถ้าทางซ้ายของ t ไม่เป็นโมฆะ −

      • แทรกด้านซ้ายของ t ลงใน q

    • ถ้าสิทธิ์ของ t ไม่เป็นโมฆะ −

      • แทรกด้านขวาของ t ลงใน q

    • (ลดลง n โดย 1)

  • ถ้าอุณหภูมิเท่ากับ 1 แล้ว −

    • แทรก temp[0] ที่ส่วนท้ายของผลลัพธ์

  • มิฉะนั้นเมื่อขนาดของอุณหภูมิ> 1 แล้ว −

    • ผลรวม :=0

    • สำหรับการเริ่มต้น i :=0 เมื่อ i

      • sum :=sum + temp[i]

    • แทรก (ผลรวม / ขนาดอุณหภูมิ) ที่ส่วนท้ายของผลลัพธ์

• ส่งคืนผลลัพธ์

ตัวอย่าง

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

#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<float> averageOfLevels(TreeNode *root){
      vector<float> result;
      queue<TreeNode*> q;
      q.push(root);
      while (!q.empty()) {
         int n = q.size();
         vector<float> temp;
         while (n) {
            TreeNode* t = q.front();
            temp.push_back(t->val);
            q.pop();
            if (t->left && t->left->val != 0)
               q.push(t->left);
            if (t->right && t->right->val != 0)
               q.push(t->right);
               n--;
         }
         if (temp.size() == 1)
            result.push_back(temp[0]);
         else if (temp.size() > 1) {
            double sum = 0;
            for (int i = 0; i < temp.size(); i++) {
               sum += temp[i];
            }
            result.push_back(sum / temp.size());
         }
      }
      return result;
   }
};
main(){
   Solution ob;
   vector<int> v = {3,9,20,NULL,NULL,15,7};
   TreeNode *root = make_tree(v);
   print_vector(ob.averageOfLevels(root));
}

อินพุต

{3,9,20,NULL,NULL,15,7}

ผลลัพธ์

[3, 14.5, 11, ]