/* Program to find LCA of n1 and n2 using one traversal of Binary Tree By Gohired.in */
#include <iostream>
using namespace std;
struct Node
{
struct Node *left, *right;
int key;
};
Node* newNode(int key)
{
Node *temp = new Node;
temp->key = key;
temp->left = temp->right = NULL;
return temp;
}
struct Node *findLCA(struct Node* root, int n1, int n2)
{
if (root == NULL) return NULL;
if (root->key == n1 || root->key == n2)
return root;
Node *left_lca = findLCA(root->left, n1, n2);
Node *right_lca = findLCA(root->right, n1, n2);
if (left_lca && right_lca) return root;
return (left_lca != NULL)? left_lca: right_lca;
}
int main()
{
Node * root = newNode(1);
root->left = newNode(2);
root->right = newNode(3);
root->left->left = newNode(4);
root->left->right = newNode(5);
root->right->left = newNode(6);
root->right->right = newNode(7);
cout << "LCA(4, 5) = " << findLCA(root, 4, 5)->key<<endl;
cout << "LCA(4, 6) = " << findLCA(root, 4, 6)->key<<endl;
cout << "LCA(3, 4) = " << findLCA(root, 3, 4)->key<<endl;
cout << "LCA(2, 4) = " << findLCA(root, 2, 4)->key<<endl;
return 0;
}
LyogUHJvZ3JhbSB0byBmaW5kIExDQSBvZiBuMSBhbmQgbjIgdXNpbmcgb25lIHRyYXZlcnNhbCBvZiBCaW5hcnkgVHJlZSBCeSBHb2hpcmVkLmluICovCiNpbmNsdWRlIDxpb3N0cmVhbT4KdXNpbmcgbmFtZXNwYWNlIHN0ZDsKCnN0cnVjdCBOb2RlCnsKCXN0cnVjdCBOb2RlICpsZWZ0LCAqcmlnaHQ7CglpbnQga2V5Owp9OwoKTm9kZSogbmV3Tm9kZShpbnQga2V5KQp7CglOb2RlICp0ZW1wID0gbmV3IE5vZGU7Cgl0ZW1wLT5rZXkgPSBrZXk7Cgl0ZW1wLT5sZWZ0ID0gdGVtcC0+cmlnaHQgPSBOVUxMOwoJcmV0dXJuIHRlbXA7Cn0KCnN0cnVjdCBOb2RlICpmaW5kTENBKHN0cnVjdCBOb2RlKiByb290LCBpbnQgbjEsIGludCBuMikKewoJaWYgKHJvb3QgPT0gTlVMTCkgcmV0dXJuIE5VTEw7CgoJaWYgKHJvb3QtPmtleSA9PSBuMSB8fCByb290LT5rZXkgPT0gbjIpCgkJcmV0dXJuIHJvb3Q7CgoJTm9kZSAqbGVmdF9sY2EgPSBmaW5kTENBKHJvb3QtPmxlZnQsIG4xLCBuMik7CglOb2RlICpyaWdodF9sY2EgPSBmaW5kTENBKHJvb3QtPnJpZ2h0LCBuMSwgbjIpOwoJaWYgKGxlZnRfbGNhICYmIHJpZ2h0X2xjYSkgcmV0dXJuIHJvb3Q7CgoJcmV0dXJuIChsZWZ0X2xjYSAhPSBOVUxMKT8gbGVmdF9sY2E6IHJpZ2h0X2xjYTsKfQoKaW50IG1haW4oKQp7CglOb2RlICogcm9vdCA9IG5ld05vZGUoMSk7Cglyb290LT5sZWZ0ID0gbmV3Tm9kZSgyKTsKCXJvb3QtPnJpZ2h0ID0gbmV3Tm9kZSgzKTsKCXJvb3QtPmxlZnQtPmxlZnQgPSBuZXdOb2RlKDQpOwoJcm9vdC0+bGVmdC0+cmlnaHQgPSBuZXdOb2RlKDUpOwoJcm9vdC0+cmlnaHQtPmxlZnQgPSBuZXdOb2RlKDYpOwoJcm9vdC0+cmlnaHQtPnJpZ2h0ID0gbmV3Tm9kZSg3KTsKCWNvdXQgPDwgIkxDQSg0LCA1KSA9ICIgPDwgZmluZExDQShyb290LCA0LCA1KS0+a2V5PDxlbmRsOwoJY291dCA8PCAiTENBKDQsIDYpID0gIiA8PCBmaW5kTENBKHJvb3QsIDQsIDYpLT5rZXk8PGVuZGw7Cgljb3V0IDw8ICJMQ0EoMywgNCkgPSAiIDw8IGZpbmRMQ0Eocm9vdCwgMywgNCktPmtleTw8ZW5kbDsKCWNvdXQgPDwgIkxDQSgyLCA0KSA9ICIgPDwgZmluZExDQShyb290LCAyLCA0KS0+a2V5PDxlbmRsOwoJcmV0dXJuIDA7Cn0K