二叉树的层序遍历
1、给一个二叉树,请返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
广度优先遍历BFS:
/**
* 二叉树的层序遍历:队列(先出先进,每次入队一层,出队一层的节点):广度优先遍历
* 时间复杂度:O(n)
* 空间复杂度:O(n)
* @param root
* @return
*/
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if(root == null){
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
//根节点入队列
queue.add(root);
while (queue.size() >0){
//队列中还有元素
//获取队列元素的个数,队列元素中的个数相当于该层的节点数量
int number = queue.size();
List<Integer> temp = new ArrayList<>();
//将队列中节点(该层的所以有节点)出队,放入temp集合中,
// 如果节点的左右子树不为空,并将孩子节点(下层的节点)入队
for(int i = 0 ;i<number;i++){
//出队,将节点从队列弹出
TreeNode tree = queue.poll();
temp.add(tree.val);
if(tree.left != null){
queue.add(tree.left);
}
if(tree.right != null){
queue.add(tree.right);
}
}
//将temp加入最终返回结果中
res.add(temp);
}
return res;
}
深度优先遍历DFS:
/**
* 深度优先遍历dfs: DFS 不是按照层次遍历的,
* 为了让递归的过程中同一层的节点放到同一个列表中,在递归时要记录每个节点的深度 level。
* 递归到新节点要把该节点放入 level 对应列表的末尾。
* @param res
* @param node
* @param level
* @return
*/
private void dfs(List<List<Integer>> res,TreeNode node,int level){
//当遍历到一个新的深度 level,而最终结果 res 中还没有创建 level 对应的列表时,
// 应该在 res 中新建一个列表用来保存该 level 的所有节点。
if(res.size()-1<level){
res.add(new ArrayList<>());
}
//把该节点放入level对应列表的末尾
res.get(level).add(node.val);
if(node.left != null){
dfs(res,node.left,level+1);
}
if(node.right != null){
dfs(res,node.right,level+1);
}
}
/**
* 二叉树的层次遍历:深度优先遍历DFS:递归
* @param root
* @return
*/
public List<List<Integer>> levelOrder2(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if(root != null){
dfs(res,root,0);
}
return res;
}