-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathBinaryTree.java
92 lines (75 loc) · 2.45 KB
/
BinaryTree.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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
public class BinaryTree {
public BinaryNode root;
public BinaryTree() {}
public BinaryTree(String value) {
this.root = new BinaryNode(value);
}
public BinaryNode addRoot(String value) {
this.root = new BinaryNode(value);
return root;
}
public BinaryNode addLeftChild(BinaryNode node, String value) throws Exception {
if (node.left == null) {
BinaryNode newNode = new BinaryNode(value, node);
node.left = newNode;
return newNode;
} else {
throw new Exception("Node already have left child");
}
}
public BinaryNode addRightChild(BinaryNode node, String value) throws Exception {
if (node.right == null) {
BinaryNode newNode = new BinaryNode(value, node);
node.right = newNode;
return newNode;
} else {
throw new Exception("Node already have right child");
}
}
public int getNodesCount() {
return root.getChildrenCount() + 1;
}
public int getExternalNodesCount() {
return root.getExternalChildrenCount();
}
public int getLevelsCount() {
return countLevels(root, 0);
}
private int countLevels(BinaryNode node, int parentLevel) {
int leftLevel = parentLevel;
int rightLevel = parentLevel;
if (node.left != null) leftLevel = countLevels(node.left, leftLevel + 1);
if (node.right != null) rightLevel = countLevels(node.right, rightLevel + 1);
return (leftLevel > rightLevel ? leftLevel : rightLevel);
}
public String printPreOrder() {
return printPreOrder(root);
}
private String printPreOrder(BinaryNode node) {
String print = "";
print += "[" + node.value + "]";
if (node.left != null) print += " - " + printPreOrder(node.left);
if (node.right != null) print += " - " + printPreOrder(node.right);
return print;
}
public String printPosOrder() {
return printPosOrder(root);
}
private String printPosOrder(BinaryNode node) {
String print = "";
if (node.left != null) print += printPosOrder(node.left) + " - ";
if (node.right != null) print += printPosOrder(node.right) + " - ";
print += "[" + node.value + "]";
return print;
}
public String printInterOrder() {
return printInterOrder(root);
}
public String printInterOrder(BinaryNode node) {
String print = "";
if (node.left != null) print += printInterOrder(node.left) + " - ";
print += "[" + node.value + "]";
if (node.right != null) print += " - " + printInterOrder(node.right);
return print;
}
}