102.二叉樹的層序遍歷
- 二叉樹的層序遍歷
難度中等460
給你一個二叉樹,請你返回其按 層序遍歷 得到的節點值。 (即逐層地,從左到右訪問所有節點)。
示例:
二叉樹:[3,9,20,null,null,15,7],
3 / \ 9 20 / \ 15 7
返回其層次遍歷結果:
[ [3], [9,20], [15,7] ]
1.DFS
List<List<Integer>> result = new ArrayList<>();
//1.DFS 將每一次用k記錄 添加到對應的層級
// time :O(n)
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null) return result;
recur(root,0);
return result;
}
private void recur(TreeNode root,int k){
if(root != null){
if(result.size()<=k){
result.add(new ArrayList<>());
}
result.get(k).add(root.val);
recur(root.left,k+1);
recur(root.right,k+1);
}
}
2.BFS
//2.BFS 層序遍歷 將每層節點添加到queue中
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> result = new ArrayList<>();
if(root == null) return result;
Queue<TreeNode> queue = new LinkedList<>();
//1.添加首節點
queue.add(root);
while(!queue.isEmpty()){
int cnt = queue.size();
List<Integer> list = new ArrayList<>();
while(cnt-->0){
TreeNode node = queue.poll();
if(node == null){
continue;
}
list.add(node.val);
queue.add(node.left);
queue.add(node.right);
}
if(queue.size()!=0){
result.add(list);
}
}
return result;
}