C++の各ツリー行で最大値を見つける

二分木があるとすると、その木の各レベルの最大の要素を見つける必要があります。したがって、ツリーが次のような場合-

その場合、出力は[3,5,8]

これを解決するには、次の手順に従います-

• ans

  という配列を定義します

• 再帰関数solve（）を定義します。これはツリーノードを取り、レベルは最初は0です。このメソッドは-

  のように機能します。

• ノードがnullの場合は、

  を返します。

• level =ansのサイズの場合、ノード値をansに挿入します。それ以外の場合、ans [level]：=ans[level]とノード値の最大値

• 呼び出しsolve（ノードの左側のサブツリー、レベル+ 1）

• 呼び出しsolve（ノードの右側のサブツリー、レベル+ 1）

• mainメソッドから、rootをパラメーターとして使用してsolve（）を呼び出し、level =0

• 次に、ansを返します

理解を深めるために、次の実装を見てみましょう-

例

#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, ]