Skip to content

Instantly share code, notes, and snippets.

@yaki29
Created June 9, 2018 15:50
Show Gist options
  • Save yaki29/bffd436ce1c35de7043e200c671fa97a to your computer and use it in GitHub Desktop.
Save yaki29/bffd436ce1c35de7043e200c671fa97a to your computer and use it in GitHub Desktop.
#include <bits/stdc++.h>
using namespace std;
/* A binary tree node has data, pointer to left child
and a pointer to right child */
struct Node
{
int data;
struct Node* left;
struct Node* right;
};
Node * LCA(Node* ,int ,int );
/* Helper function that allocates a new node with the
given data and NULL left and right pointers. */
struct Node* newNode(int data)
{
struct Node* node = (struct Node*)
malloc(sizeof(struct Node));
node->data = data;
node->left = NULL;
node->right = NULL;
return(node);
}
/* Driver program to test size function*/
int main()
{
int t;
struct Node *child;
scanf("%d", &t);
while (t--)
{
map<int, Node*> m;
int n;
scanf("%d",&n);
struct Node *root = NULL;
if(n==1)
{
int a;
cin>>a;
cout<<a<<endl;
}else{
while (n--)
{
Node *parent;
char lr;
int n1, n2;
scanf("%d %d %c", &n1, &n2, &lr);
// cout << n1 << " " << n2 << " " << (char)lr << endl;
if (m.find(n1) == m.end())
{
parent = newNode(n1);
m[n1] = parent;
if (root == NULL)
root = parent;
}
else
parent = m[n1];
child = newNode(n2);
if (lr == 'L')
parent->left = child;
else
parent->right = child;
m[n2] = child;
}
int a,b;
cin>>a>>b;
Node * k = LCA(root,a,b);
cout<<k->data<<endl;
}
}
return 0;
}
/*Please note that it's Function problem i.e.
you need to write your solution in the form of Function(s) only.
Driver Code to call/invoke your function is mentioned above.*/
/* A binary tree node
struct Node
{
int data;
Node* left, * right;
}; */
/*you are required to
complete this function */
deque<Node*> dq;
vector<Node*> path, path1;
void preorder(Node* root, int ghk)
{
if(root == NULL)
return;
dq.push_back(root);
if(root->data == ghk)
{
while(!dq.empty())
{
Node* x = dq.front();
path.push_back(x);
dq.pop_front();
}
// return;
}
// cout<<root->data<<" ";
preorder(root->left, ghk);
preorder(root->right, ghk);
if(!dq.empty())
dq.pop_back();
}
Node * LCA(Node* root ,int n1 ,int n2 )
{
//Your code here
preorder(root, n1);
cout<<endl;
// path1 = path;
for(int i = 0; i<path.size(); i++)
cout<<path[i]->data<<" ";
cout<<endl;
path.clear();
dq.clear();
preorder(root, n2);
cout<<endl;
// Node* ans = NULL;
// for(int i = 0; i<path.size(); i++)
// cout<<path[i]->data<<" ";
// cout<<endl;
// for(int i = 0; i<path1.size(); i++)
// cout<<path1[i]->data<<" ";
// cout<<endl;
// for(int i = 0; i<min(path.size(), path1.size()); i++)
// {
// if(path[i]->data == path1[i]->data)
// {
// ans = path[i];
// }
// }
return root;
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment