이진 트리가 있다고 가정하고 가장 깊은 잎의 값의 합을 찾아야 합니다. 트리가 다음과 같다면 -
그러면 출력은 15가 됩니다.
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
- 지도 m 및 maxDepth 정의
- 재귀적 방법 solve()를 정의하면 노드와 레벨이 필요하며 초기 레벨은 0입니다.
- 노드가 없으면 반환
- maxDepth :=레벨 및 maxDepth의 최대값
- 노드 값만큼 m[레벨] 증가
- 해결(노드 왼쪽, 레벨 + 1)
- 해결(노드 오른쪽, 레벨 + 1)
- 메인 메소드에서 maxDepth :=0으로 설정한 다음 solve(root, 0)
- m[maxDepth] 반환
예시(C++)
더 나은 이해를 위해 다음 구현을 살펴보겠습니다. −
#include <bits/stdc++.h> using namespace std; 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: int maxDepth; map <int, int> m; void solve(TreeNode* node, int level = 0){ if(!node)return; maxDepth = max(level, maxDepth); m[level] += node->val; solve(node->left, level + 1); solve(node->right, level + 1); } int deepestLeavesSum(TreeNode* root) { maxDepth = 0; m.clear(); solve(root); //cout << maxDepth << endl; return m[maxDepth]; } }; main(){ vector<int> v = {1,2,3,4,5,NULL,6,7,NULL,NULL,NULL,NULL,8}; TreeNode *root = make_tree(v); Solution ob; cout << (ob.deepestLeavesSum(root)); }
입력
[1,2,3,4,5,null,6,7,null,null,null,null,8]
출력
15