-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #10 from Rajanvik/patch-1
Update LowestCommonAncestor.java
- Loading branch information
Showing
1 changed file
with
132 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1 +1,133 @@ | ||
//{ Driver Code Starts | ||
import java.util.LinkedList; | ||
import java.util.Queue; | ||
import java.io.*; | ||
import java.util.*; | ||
|
||
class Node{ | ||
int data; | ||
Node left; | ||
Node right; | ||
Node(int data){ | ||
this.data = data; | ||
left=null; | ||
right=null; | ||
} | ||
} | ||
|
||
class LowestCommonAncestor { | ||
|
||
static Node buildTree(String str){ | ||
|
||
if(str.length()==0 || str.charAt(0)=='N'){ | ||
return null; | ||
} | ||
|
||
String ip[] = str.split(" "); | ||
// Create the root of the tree | ||
Node root = new Node(Integer.parseInt(ip[0])); | ||
// Push the root to the queue | ||
|
||
Queue<Node> queue = new LinkedList<>(); | ||
|
||
queue.add(root); | ||
// Starting from the second element | ||
|
||
int i = 1; | ||
while(queue.size()>0 && i < ip.length) { | ||
|
||
// Get and remove the front of the queue | ||
Node currNode = queue.peek(); | ||
queue.remove(); | ||
|
||
// Get the current node's value from the string | ||
String currVal = ip[i]; | ||
|
||
// If the left child is not null | ||
if(!currVal.equals("N")) { | ||
|
||
// Create the left child for the current node | ||
currNode.left = new Node(Integer.parseInt(currVal)); | ||
// Push it to the queue | ||
queue.add(currNode.left); | ||
} | ||
|
||
// For the right child | ||
i++; | ||
if(i >= ip.length) | ||
break; | ||
|
||
currVal = ip[i]; | ||
|
||
// If the right child is not null | ||
if(!currVal.equals("N")) { | ||
|
||
// Create the right child for the current node | ||
currNode.right = new Node(Integer.parseInt(currVal)); | ||
|
||
// Push it to the queue | ||
queue.add(currNode.right); | ||
} | ||
i++; | ||
} | ||
|
||
return root; | ||
} | ||
static void printInorder(Node root) | ||
{ | ||
if(root == null) | ||
return; | ||
|
||
printInorder(root.left); | ||
System.out.print(root.data+" "); | ||
|
||
printInorder(root.right); | ||
} | ||
|
||
public static void main (String[] args) throws IOException{ | ||
BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); | ||
|
||
int t=Integer.parseInt(br.readLine()); | ||
|
||
while(t-- > 0){ | ||
String input[] = br.readLine().trim().split(" "); | ||
int a = Integer.parseInt(input[0]); | ||
int b = Integer.parseInt(input[1]); | ||
String s = br.readLine(); | ||
Node root = buildTree(s); | ||
Solution g = new Solution(); | ||
Node k = g.lca(root,a,b); | ||
System.out.println(k.data); | ||
|
||
} | ||
} | ||
} | ||
|
||
|
||
class Solution | ||
|
||
{ | ||
|
||
//Function to return the lowest common ancestor in a Binary Tree. | ||
|
||
Node lca(Node root, int n1,int n2) | ||
|
||
{ | ||
|
||
|
||
if(root == null || root.data == n1 || root.data == n2) return root; | ||
|
||
Node left = lca(root.left,n1,n2); | ||
|
||
Node right = lca(root.right,n1,n2); | ||
|
||
if(left == null) return right; | ||
|
||
if(right == null) return left; | ||
|
||
else return root; | ||
|
||
|
||
} | ||
|
||
} |