-
Notifications
You must be signed in to change notification settings - Fork 111
/
solution.java
68 lines (61 loc) · 1.86 KB
/
solution.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import java.util.*;
import java.io.*;
class Node {
Node left;
Node right;
int data;
Node(int data) {
this.data = data;
left = null;
right = null;
}
}
class solution {
//Method to calculate the height of the tree
//We compute the height of the tree by recursively traversing
//the left subtree and the right subtree
public static int height(Node root) {
// Set the leftHeight and rightHeight of current node to zero
int leftHeight = 0;
int rightHeight = 0;
//If left of current node exists increase left height by 1
if(root.left != null)
{
leftHeight = 1 + height(root.left);
}
//If right of current node exists increase rightheight by 1
if(root.right != null)
{
rightHeight = 1 + height(root.right);
}
//Return the maximum value in between leftHeight and rightHeight
return Math.max(leftHeight, rightHeight);
}
public static Node insert(Node root, int data) {
if(root == null) {
return new Node(data);
} else {
Node cur;
if(data <= root.data) {
cur = insert(root.left, data);
root.left = cur;
} else {
cur = insert(root.right, data);
root.right = cur;
}
return root;
}
}
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int t = scan.nextInt();
Node root = null;
while(t-- > 0) {
int data = scan.nextInt();
root = insert(root, data);
}
scan.close();
int height = height(root);
System.out.println(height);
}
}