二叉树的层序遍历
方法:广度优先搜索
题目描述:给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:二叉树:[3,9,20,null,null,15,7],
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
思路:由于要求输出的结果返回一个二维数组,把每一层单独输出,因此在变量的过程中需要对每一层区分。可以在每一层遍历前记录队列中的结点数量,然后进行遍历这一层。
代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(root == null) return res;
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.add(root); //还可以用queue.offer(root);
while(!queue.isEmpty()){
List<Integer> level = new ArrayList<Integer>();
int levelsize = queue.size();
for(int i=1;i<=levelsize;i++){
TreeNode node = queue.poll(); //java中pop()写作poll
level.add(node.val);
if(node.left!=null)
queue.add(node.left);
id(node.right!=null)
queue.add(node,right);
}
res.add(level);
}
return res;
}
}
题目链接:
102.二叉树的层序遍历
参考