พิจารณาว่าเรามีไบนารีทรีที่มีโหนดไม่กี่โหนด เราต้องหาระยะห่างระหว่างรูทกับโหนดอื่น สมมติว่าต้นไม้มีลักษณะดังนี้:
ตอนนี้ระยะห่างระหว่าง (รูท, 6) =2, ความยาวเส้นทางคือ 2, ระยะห่างระหว่าง (รูท, 8) =3 เป็นต้น
ในการแก้ปัญหานี้ เราจะใช้วิธีการแบบเรียกซ้ำเพื่อค้นหาโหนดที่ทรีย่อยด้านซ้ายและขวา และอัปเดตความยาวสำหรับแต่ละระดับด้วย
ตัวอย่าง
#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;
}
int getDistance(Node *root, int x) {
if (root == NULL)
return -1;
int dist = -1;
if ((root->data == x) || (dist = getDistance(root->left, x)) >= 0 || (dist = getDistance(root->right, x)) >= 0)
return dist + 1;
return dist;
}
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 from root to node 6 is: " << getDistance(root,6);
cout << "\nDistance from root to node 8 is: " << getDistance(root,8);
} ผลลัพธ์
Distance from root to node 6 is: 2 Distance from root to node 8 is: 3
