이진 트리가 있다고 가정하고 오른쪽에서 트리를 보면 일부 요소를 볼 수 있습니다. 이러한 요소를 표시해야 합니다. 트리가 다음과 같다면 -
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
- dfs를 위한 하나의 도움 방법을 만들 것입니다. 이것은 tree_node, 답변을 보유하는 배열 및 레벨을 취합니다. 레벨은 초기에 0입니다. dfs는 아래와 같이 작동합니다 -
- 노드가 null이면 반환
- 레벨 =응답 배열의 길이인 경우 노드의 값을 as 배열에 삽입
- dfs(노드 오른쪽, ans, 레벨 + 1)
- dfs(노드 왼쪽, ans, 레벨 + 1)
- 메인 함수에서 트리의 루트와 하나의 빈 배열을 사용하여 dfs()를 호출합니다. 레벨은 초기에 0이므로 dfs(root, ans)가 됩니다.
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
예시
#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; } }; 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: void dfs(TreeNode* node, vector <int>& ans, int level = 0){ if(!node) return; if(level == ans.size())ans.push_back(node->val); dfs(node->right, ans, level + 1); dfs(node->left, ans, level + 1); } vector<int> rightSideView(TreeNode* root) { vector <int> ans; dfs(root, ans); return ans; } }; main(){ vector<int> v = {1,2,3,NULL,5,NULL,4}; TreeNode *root = make_tree(v); Solution ob; print_vector(ob.rightSideView(root)); }
입력
[1,2,3,null,5,null,4]
출력
[1, 3, 4, ]