Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

C++에서 이진 트리의 두 노드 사이의 거리 찾기


노드가 거의 없는 이진 트리가 있다고 가정해 보겠습니다. 우리는 두 노드 u와 v 사이의 거리를 찾아야 합니다. 트리가 아래와 같다고 가정합니다 -

C++에서 이진 트리의 두 노드 사이의 거리 찾기

이제 (4, 6) =4 사이의 거리, 경로 길이는 4, (5, 8) =5 사이의 길이 등

이 문제를 해결하기 위해 LCA(Lowest Common Ancestor)를 찾은 다음 LCA에서 두 노드까지의 거리를 계산합니다.

예시

#include<iostream>
using namespace std;
class Node {
   public:
      int data;
   Node *left, *right;
};
Node* getNode(int data) {
   Node* node = new Node;
   node->data = data;
   node->left = node->right = NULL;
   return node;
}
Node* LowestCommonAncestor(Node * root, int n1,int n2) {
   if (root == NULL)
      return root;
   if (root->data == n1 || root->data == n2)
      return root;
   Node* left = LowestCommonAncestor(root->left, n1, n2);
   Node* right = LowestCommonAncestor(root->right, n1, n2);
   if (left != NULL && right != NULL)
      return root;
   if (left != NULL)
      return LowestCommonAncestor(root->left, n1, n2);
   return LowestCommonAncestor(root->right, n1, n2);
}
int getLevel(Node *root, int k, int level) {
   if(root == NULL) return -1;
      if(root->data == k) return level;
         int left = getLevel(root->left, k, level+1);
      if (left == -1)
         return getLevel(root->right, k, level+1);
   return left;
}
int findDistance(Node* root, int a, int b) {
   Node* lca = LowestCommonAncestor(root, a , b);
   int dist1 = getLevel(lca, a, 0);
   int dist2 = getLevel(lca, b, 0);
   return dist1 + dist2;
}
int main() {
   Node* root = getNode(1);
   root->left = getNode(2);
   root->right = getNode(3);
   root->left->left = getNode(4);
   root->left->right = getNode(5);
   root->right->left = getNode(6);
   root->right->right = getNode(7);
   root->right->left->right = getNode(8);
   cout << "Distance between (4, 6) is: " << findDistance(root, 4, 6);
   cout << "\nDistance between (8, 5) is: " << findDistance(root, 8, 5);
}

출력

Distance between (4, 6) is: 4
Distance between (8, 5) is: 5