C++에서 이진 트리의 왼쪽 보기를 찾는 프로그램

이진 트리가 있다고 가정하고 왼쪽에서 트리를 보면 일부 요소를 볼 수 있습니다. 이러한 요소를 표시해야 합니다. 트리가 다음과 같다면 -

출력은 [1,2,5]

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

• ret 배열 정의

• dfs() 함수를 정의하면 노드가 필요하고 c는 1로 초기화합니다.

• 노드가 null이면 -

  • 반환

• c> lvl이면 -

  • 레벨 :=c

  • 노드의 값을 ret에 삽입

• dfs(노드의 왼쪽, c + 1)

• dfs(노드 오른쪽, c + 1)

• 기본 방법에서 다음을 수행하십시오 -

• 레벨 :=-1

• dfs(루트, 0)

• 리턴 렛

  
  

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

예시

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<int> 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 = right = NULL;
   }
};
class Solution {
   public:
   vector <int> ret;
   int lvl;
   void dfs(TreeNode* node, int c = 1){
      if(!node)
         return;
      if(c > lvl){
         lvl = c;
         ret.push_back(node->val);
      }
      dfs(node->left, c + 1);
      dfs(node->right, c + 1);
   }
   vector<int> solve(TreeNode* root) {
      lvl = -1;
      dfs(root, 0);
      return ret;
   }
};
main(){
   TreeNode *root = new TreeNode(1);
   root->left = new TreeNode(2);
   root->right = new TreeNode(3);
   root->left->right = new TreeNode(5);
   root->right->right = new TreeNode(4);
   Solution ob;
   print_vector(ob.solve(root));
}

입력

TreeNode *root = new TreeNode(1);
root->left = new TreeNode(2);
root->right = new TreeNode(3);
root->left->right = new TreeNode(5);
root->right->right = new TreeNode(4);

출력

[1,2,5]