-
Notifications
You must be signed in to change notification settings - Fork 200
/
Copy pathTraverser.java
53 lines (41 loc) · 1.04 KB
/
Traverser.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
package Trees.LevelOrderTraversal;
import java.util.LinkedList;
import java.util.Queue;
class TreeNode {
int value;
TreeNode left;
TreeNode right;
public TreeNode(int value) {
this.value = value;
this.left = null;
this.right = null;
}
}
public class Traverser {
public static void levelOrder(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
// Counters for printing a "\n" after the current level
int currentLevel = 1, nextLevel = 0;
while (!queue.isEmpty()) {
TreeNode tempNode = queue.poll();
currentLevel--;
System.out.print(tempNode.value + " ");
if (tempNode.left != null) {
queue.add(tempNode.left);
nextLevel++;
}
if (tempNode.right != null) {
queue.add(tempNode.right);
nextLevel++;
}
// If currentLevel is 0, that means the next
// level will start
if (currentLevel == 0) {
System.out.println();
currentLevel = nextLevel;
nextLevel = 0;
}
}
}
}