-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathLevelOrderTraversal.java
More file actions
39 lines (28 loc) · 1.01 KB
/
LevelOrderTraversal.java
File metadata and controls
39 lines (28 loc) · 1.01 KB
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
import java.util.*;
class TreeNode {
int val;
TreeNode left, right;
TreeNode(int x) { val = x; }
}
public class LevelOrderTraversal {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList<>();
if (root == null) return result;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root); // Start with root
while (!queue.isEmpty()) {
int levelSize = queue.size(); // Number of nodes at current level
List<Integer> level = new ArrayList<>();
for (int i = 0; i < levelSize; i++) {
TreeNode current = queue.poll(); // Remove from queue
level.add(current.val);
if (current.left != null)
queue.offer(current.left);
if (current.right != null)
queue.offer(current.right);
}
result.add(level); // Add this level to final result
}
return result;
}
}