이진 트리가 있다고 가정합니다. 가장 긴 연속 시퀀스 경로의 길이를 찾을 수 있는지 확인해야 합니다. 경로가 일부 시작 노드에서 상위-하위 연결을 따라 트리의 임의 노드까지의 노드 시퀀스를 참조하는 경우. 가장 긴 연속 경로는 부모에서 자식으로 따라야 하지만 반대로는 안 됩니다.
따라서 입력이 다음과 같으면
가장 긴 연속 시퀀스 경로가 3-4-5이므로 출력은 3이 되므로 3을 반환합니다.
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
-
solveUtil() 함수를 정의하면 node, prev, len이 1로 초기화됩니다.
-
노드가 null이면 -
-
반환
-
-
prev + 1이 노드의 val과 같으면 -
-
(len 1씩 증가)
-
ans :=ans 및 len의 최대값
-
solveUtil(노드의 왼쪽, 노드의 val, len)
-
solveUtil(노드의 오른쪽, 노드의 val, len)
-
-
그렇지 않으면
-
solveUtil(노드의 왼쪽, 노드의 값, 1)
-
solveUtil(노드의 오른쪽, 노드의 val, 1)
-
-
solve() 함수를 정의하면 A가 필요합니다.
-
답변 :=1
-
solveUtil(A, -무한대)
-
반환
-
주요 방법에서 다음을 수행하십시오 -
-
루트가 null이면 -
-
0 반환
-
-
해결(루트) 반환
예시
더 나은 이해를 위해 다음 구현을 살펴보겠습니다. −
#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 ans; void solveUtil(TreeNode* node, int prev, int len = 1){ if (!node) return; if (prev + 1 == node->val) { len++; ans = max(ans, len); solveUtil(node->left, node->val, len); solveUtil(node->right, node->val, len); } else { solveUtil(node->left, node->val, 1); solveUtil(node->right, node->val, 1); } } int solve(TreeNode* A){ ans = 1; solveUtil(A, INT_MIN); return ans; } int longestConsecutive(TreeNode* root){ if (!root) return 0; return solve(root); } }; main(){ Solution ob; TreeNode *root = new TreeNode(1); root->right = new TreeNode(3); root->right->left = new TreeNode(2); root->right->right = new TreeNode(4); root->right->right->right = new TreeNode(5); cout << (ob.longestConsecutive(root)); }
입력
TreeNode *root = new TreeNode(1); root->right = new TreeNode(3); root->right->left = new TreeNode(2); root->right->right = new TreeNode(4); root->right->right->right = new TreeNode(5);
출력
3